"vscode:/vscode.git/clone" did not exist on "4f882be4a0d319551ce2a0eadcace0e76f3432cd"
Unverified Commit 6e2c176e authored by Michael Goin's avatar Michael Goin Committed by GitHub
Browse files

[Bugfix] Restrict Machete to only run on Hopper (#20830)


Signed-off-by: default avatarmgoin <mgoin64@gmail.com>
parent a86754a1
...@@ -32,6 +32,9 @@ class MacheteLinearKernel(MPLinearKernel): ...@@ -32,6 +32,9 @@ class MacheteLinearKernel(MPLinearKernel):
if not current_platform.is_cuda(): if not current_platform.is_cuda():
return False, "Machete only supported on CUDA" return False, "Machete only supported on CUDA"
if not current_platform.is_device_capability(90):
return False, "Machete requires compute capability of 90 (Hopper)"
if c.has_g_idx and\ if c.has_g_idx and\
c.partition_weight_shape[0] != c.full_weight_shape[0]: c.partition_weight_shape[0] != c.full_weight_shape[0]:
return False, "Act reordering currently not supported by Machete, "\ return False, "Act reordering currently not supported by Machete, "\
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment