From 4b15723e59f8006f855042ec97ccc4f11f659f72 Mon Sep 17 00:00:00 2001
From: Zhuohan Li <zhuohan123@vip.qq.com>
Date: Wed, 20 Mar 2024 23:54:35 +0000
Subject: [PATCH] [BugFix] Hot fix in setup.py for neuron build

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 67575a0e04bf0..47cac5996f816 100644
--- a/setup.py
+++ b/setup.py
@@ -168,7 +168,7 @@ def build_extensions(self) -> None:
 
 
 def _is_cuda() -> bool:
-    return torch.version.cuda is not None
+    return torch.version.cuda is not None and not _is_neuron()
 
 
 def _is_hip() -> bool: