Try enable test_moe_matmul_ogs on CI (#147)

yf225 · web-flow · commit 0e047afc7cc5 · 2025-06-10T11:31:42.000+02:00
diff --git a/test/test_examples.py b/test/test_examples.py
@@ -1628,8 +1628,8 @@ def _jagged_dense_add_2d_make_precompiler(x_data: torch.Tensor, x_offsets: torch
         )
 
     @unittest.skipIf(
-        torch.cuda.get_device_capability(0) < (9, 0),
-        "Triton internal error on RTX 3090",
+        "RTX 30" in torch.cuda.get_device_name(0),
+        "Triton internal error on RTX 30XX series",
     )
     @unittest.skipIf(is_fbcode(), "Triton internal error on fbcode Triton pin")
     def test_moe_matmul_ogs(self):

Original file line number	Diff line number	Diff line change
`@@ -1628,8 +1628,8 @@ def _jagged_dense_add_2d_make_precompiler(x_data: torch.Tensor, x_offsets: torch`
`1628`	`1628`	`)`
`1629`	`1629`
`1630`	`1630`	`@unittest.skipIf(`
`1631`		`- torch.cuda.get_device_capability(0) < (9, 0),`
`1632`		`- "Triton internal error on RTX 3090",`
	`1631`	`+ "RTX 30" in torch.cuda.get_device_name(0),`
	`1632`	`+ "Triton internal error on RTX 30XX series",`
`1633`	`1633`	`)`
`1634`	`1634`	`@unittest.skipIf(is_fbcode(), "Triton internal error on fbcode Triton pin")`
`1635`	`1635`	`def test_moe_matmul_ogs(self):`