From 429e5fb868376f9aed889f519fe238135362f109 Mon Sep 17 00:00:00 2001
From: Leyuan Wang <laurawly@gmail.com>
Date: Fri, 1 Jun 2018 10:54:06 -0700
Subject: [PATCH] [Metal] Fix block launching parameter (#1219)

---
 src/runtime/metal/metal_module.mm | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/runtime/metal/metal_module.mm b/src/runtime/metal/metal_module.mm
index ed7cba6e75a5..5baefd72ec62 100644
--- 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];