Allow initialisation of MTLSize with tuples of different integer types (

#425)
JuliaGPU · Sep 24, 2024 · 5628d43 · 5628d43
1 parent e3d9b32
commit 5628d43
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/lib/mtl/size.jl b/lib/mtl/size.jl
@@ -11,9 +11,9 @@ struct MTLSize
 end
 
 # convenience constructors from tuple inputs
-MTLSize(dims::NTuple{1,<:Integer}) = MTLSize(dims[1], 1,       1)
-MTLSize(dims::NTuple{2,<:Integer}) = MTLSize(dims[1], dims[2], 1)
-MTLSize(dims::NTuple{3,<:Integer}) = MTLSize(dims[1], dims[2], dims[3])
+MTLSize(dims::NTuple{1,Integer}) = MTLSize(dims[1], 1,       1)
+MTLSize(dims::NTuple{2,Integer}) = MTLSize(dims[1], dims[2], 1)
+MTLSize(dims::NTuple{3,Integer}) = MTLSize(dims[1], dims[2], dims[3])
 
 
 ## origin

diff --git a/test/execution.jl b/test/execution.jl
@@ -13,13 +13,17 @@ dummy() = return
     @metal threads dummy()
     @metal threads=1 dummy()
     @metal threads=(1,1) dummy()
+    @metal threads=(Int32(1),Int64(1)) dummy()
     @metal threads=(1,1,1) dummy()
+    @metal threads=(Int8(1),Int32(1),1) dummy()
 
     groups = 1
     @metal groups dummy()
     @metal groups=1 dummy()
     @metal groups=(1,1) dummy()
+    @metal groups=(Int32(1),Int64(1)) dummy()
     @metal groups=(1,1,1) dummy()
+    @metal groups=(Int8(1),Int32(1),1) dummy()
 
     @metal macos=Metal.macos_version() dummy()
     @metal metal=Metal.metal_support() dummy()