fix lint error

chichun-charlie-liu · chichun-charlie-liu · commit 29e00bbea371 · 2025-03-17T19:41:42.000-04:00
Signed-off-by: cliu-us &lt;cliu@us.ibm.com&gt;
diff --git a/fms_mo/modules/linear.py b/fms_mo/modules/linear.py
@@ -1125,7 +1125,7 @@ def set_matmul_op(self):
                 imatmul_ops_reg,
             )
 
-            if self.use_int_kernel == "triton":
+            if self.use_int_kernel == "triton" and available_packages["triton"]:
                 # will use real imatmul written in triton
                 imm_func = partial(
                     tl_matmul,
diff --git a/tests/triton_kernels/test_triton_mm.py b/tests/triton_kernels/test_triton_mm.py
@@ -26,6 +26,8 @@
     from fms_mo.custom_ext_kernels.triton_kernels import (
         tl_matmul_chunk_truncate as tl_matmul,
     )
+else:
+    raise ImportError("triton python package is not avaialble, please check your installation.")
 
 
 @pytest.mark.parametrize("mkn", [64, 256, 1024, 4096])

Original file line number	Diff line number	Diff line change
`@@ -1125,7 +1125,7 @@ def set_matmul_op(self):`
`1125`	`1125`	`imatmul_ops_reg,`
`1126`	`1126`	`)`
`1127`	`1127`
`1128`		`- if self.use_int_kernel == "triton":`
	`1128`	`+ if self.use_int_kernel == "triton" and available_packages["triton"]:`
`1129`	`1129`	`# will use real imatmul written in triton`
`1130`	`1130`	`imm_func = partial(`
`1131`	`1131`	`tl_matmul,`
Original file line number	Diff line number	Diff line change
`@@ -26,6 +26,8 @@`
`26`	`26`	`from fms_mo.custom_ext_kernels.triton_kernels import (`
`27`	`27`	`tl_matmul_chunk_truncate as tl_matmul,`
`28`	`28`	`)`
	`29`	`+else:`
	`30`	`+ raise ImportError("triton python package is not avaialble, please check your installation.")`
`29`	`31`
`30`	`32`
`31`	`33`	`@pytest.mark.parametrize("mkn", [64, 256, 1024, 4096])`