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):
if not current_platform.is_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\
c.partition_weight_shape[0] != c.full_weight_shape[0]:
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