diff --git a/kernel-builder/src/pyproject/templates/tvm_ffi/_ops.py.in b/kernel-builder/src/pyproject/templates/tvm_ffi/_ops.py.in
index ed5eb01c..9f1fa313 100644
--- a/kernel-builder/src/pyproject/templates/tvm_ffi/_ops.py.in
+++ b/kernel-builder/src/pyproject/templates/tvm_ffi/_ops.py.in
@@ -1,7 +1,13 @@
 from pathlib import Path
+import platform
 
 import tvm_ffi
 
 _SELF_PATH = Path(__file__).parent
 
-ops = tvm_ffi.load_module(_SELF_PATH / "@OPS_NAME@.so")
+if platform.system() == "Darwin":
+    ops = tvm_ffi.load_module(_SELF_PATH / "@OPS_NAME@.dylib")
+elif platform.system() == "Windows":
+    ops = tvm_ffi.load_module(_SELF_PATH / "@OPS_NAME@.dll")
+else:
+    ops = tvm_ffi.load_module(_SELF_PATH / "@OPS_NAME@.so")