[Metal] Fix block launching parameter (apache#1219)

sergei-mironov · Aug 8, 2018 · af55324 · af55324
1 parent 753dffd
commit af55324
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/runtime/metal/metal_module.mm b/src/runtime/metal/metal_module.mm
@@ -215,7 +215,7 @@ void operator()(TVMArgs args,
     MTLSize dimGrid = MTLSizeMake(
         wl.grid_dim(0), wl.grid_dim(1), wl.grid_dim(2));
     MTLSize dimBlock = MTLSizeMake(
-        wl.block_dim(0), wl.block_dim(1), wl.work_size[2]);
+        wl.block_dim(0), wl.block_dim(1), wl.block_dim(2));
     [encoder dispatchThreadgroups: dimGrid
              threadsPerThreadgroup: dimBlock];
     [encoder endEncoding];