build: Add guard for entering triton_kernels.py

Signed-off-by: Brandon Groth <[email protected]>

Author: BrandonGroth

fms_mo/custom_ext_kernels/triton_kernels.py

@@ -15,11 +15,23 @@
 """This file contains external kernels for FP and INT8 matmul written in triton."""
 
 # Third Party
-from triton.language.extra import libdevice
 import torch
+
+# First Party
+from fms_mo.utils.import_utils import available_packages
+
+# Assume any calls to the file are requesting triton
+if not available_packages["triton"]:
+    raise ImportError(
+        "triton python package is not avaialble, please check your installation."
+    )
+
+# Third Party
+from triton.language.extra import libdevice
 import triton
 import triton.language as tl
 
+
 DTYPE_I8 = [torch.int8]
 DTYPE_F8 = [torch.float8_e4m3fn, torch.float8_e5m2]
 DTYPE_8BIT = DTYPE_I8 + DTYPE_F8