From 8cedf2992b0134fdbe82457cd0a18f8974c9fb51 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Dani=C3=ABl=20de=20Kok?= Date: Fri, 20 Mar 2026 10:22:28 +0100 Subject: [PATCH] kernel-builder: use tvm-ffi dynamic library extension by platform --- kernel-builder/src/pyproject/templates/tvm_ffi/_ops.py.in | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) 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")