From 0e18dbf076fc44de5dac4027616e9f3d9e2da45a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Dani=C3=ABl=20de=20Kok?= Date: Wed, 29 Oct 2025 12:34:53 +0000 Subject: [PATCH] Fall back to CPU if no compute framework is found --- src/kernels/utils.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/src/kernels/utils.py b/src/kernels/utils.py index 958e94e..bdadf43 100644 --- a/src/kernels/utils.py +++ b/src/kernels/utils.py @@ -65,9 +65,7 @@ def build_variant() -> str: cann_major, cann_minor = get_cann_version()[0], get_cann_version()[2] compute_framework = f"cann{cann_major}{cann_minor}" else: - raise AssertionError( - "Torch was not compiled with CUDA, Metal, XPU, NPU, or ROCm enabled." - ) + compute_framework = "cpu" torch_version = parse(torch.__version__) cpu = platform.machine()