[Metal] Fix block launching parameter (#1219)

429e5fb8 · Leyuan Wang · Tianqi Chen · b266dcfb · 429e5fb8
Commit 429e5fb8 authored Jun 01, 2018 by Leyuan Wang Committed by Tianqi Chen Jun 01, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 1 additions and 1 deletions

src/runtime/metal/metal_module.mm
+1 -1

No files found.
--- a/src/runtime/metal/metal_module.mm
+++ b/src/runtime/metal/metal_module.mm
@@ -215,7 +215,7 @@ class MetalWrappedFunc {
    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];