From 9578c1ab73ece2587c744c0bdc0beabaa3f465f0 Mon Sep 17 00:00:00 2001
From: Andrey Malyshev <elvin.nnov@gmail.com>
Date: Mon, 7 Jun 2021 19:07:21 +0300
Subject: [PATCH] Add thread_warp_size for Metal device in default target
 attributes (#8202)

---
 src/target/target_kind.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/target/target_kind.cc b/src/target/target_kind.cc
index dd579b5c8a09..b9d9706773f7 100644
--- a/src/target/target_kind.cc
+++ b/src/target/target_kind.cc
@@ -350,6 +350,7 @@ TVM_REGISTER_TARGET_KIND("opencl", kDLOpenCL)
 TVM_REGISTER_TARGET_KIND("metal", kDLMetal)
     .add_attr_option<Bool>("system-lib")
     .add_attr_option<Integer>("max_num_threads", Integer(256))
+    .add_attr_option<Integer>("thread_warp_size", Integer(16))
     .set_default_keys({"metal", "gpu"});
 
 TVM_REGISTER_TARGET_KIND("vulkan", kDLVulkan)