From 24222c674eaaac42f4c02b0f0aa597db69de1af2 Mon Sep 17 00:00:00 2001 From: David Corvoysier Date: Mon, 6 Jan 2025 13:17:12 +0100 Subject: [PATCH] fix(library): only compile CUDA extension on Linux --- optimum/quanto/library/extensions/__init__.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/optimum/quanto/library/extensions/__init__.py b/optimum/quanto/library/extensions/__init__.py index 44979e5d..f57c1057 100644 --- a/optimum/quanto/library/extensions/__init__.py +++ b/optimum/quanto/library/extensions/__init__.py @@ -12,13 +12,15 @@ # See the License for the specific language governing permissions and # limitations under the License. +import platform + import torch from .cpp import * from .extension import * -if torch.cuda.is_available(): +if torch.cuda.is_available() and platform.system() == "Linux": if torch.version.cuda: from .cuda import * elif torch.version.hip: