Skip to content

Commit 8cb946e

Browse files
authored
[TARGET] Specify correct mcpu for Metal target tags (#18822)
As per title.
1 parent 8c259a3 commit 8cb946e

1 file changed

Lines changed: 5 additions & 5 deletions

File tree

python/tvm/target/tag_registry/metal.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919
from .registry import register_tag
2020

2121

22-
def _register_metal_tag(name, max_threads, shared_mem, warp_size):
22+
def _register_metal_tag(name, max_threads, shared_mem, warp_size, mcpu):
2323
register_tag(
2424
name,
2525
{
@@ -30,12 +30,12 @@ def _register_metal_tag(name, max_threads, shared_mem, warp_size):
3030
"host": {
3131
"kind": "llvm",
3232
"mtriple": "arm64-apple-macos",
33-
"mcpu": "apple-m4",
33+
"mcpu": mcpu,
3434
},
3535
},
3636
)
3737

3838

39-
_register_metal_tag("apple/m1-gpu", 1024, 32768, 32)
40-
_register_metal_tag("apple/m1-gpu-restricted", 256, 32768, 32)
41-
_register_metal_tag("apple/m2-gpu", 1024, 32768, 32)
39+
_register_metal_tag("apple/m1-gpu", 1024, 32768, 32, "apple-m1")
40+
_register_metal_tag("apple/m1-gpu-restricted", 256, 32768, 32, "apple-m1")
41+
_register_metal_tag("apple/m2-gpu", 1024, 32768, 32, "apple-m2")

0 commit comments

Comments
 (0)