Skip to content

Commit b11a3fb

Browse files
committed
Update metal device name in metal_backend.py
1 parent dc1fd12 commit b11a3fb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

backends/apple/metal/metal_backend.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ class MetalBackend(AotiBackend):
2424

2525
@staticmethod
2626
def get_device_name() -> str:
27-
return "mps"
27+
return "metal"
2828

2929
@staticmethod
3030
def get_supported_fallback_kernels() -> Dict[str, Any]:

0 commit comments

Comments
 (0)