From 523e30ea0c5abcb447763dcd9a77b54d5c5f3239 Mon Sep 17 00:00:00 2001
From: Zhuohan Li <zhuohan123@gmail.com>
Date: Wed, 20 Mar 2024 17:59:52 -0700
Subject: [PATCH] [BugFix] Hot fix in setup.py for neuron build (#3537)

---
 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 @@ class cmake_build_ext(build_ext):
 
 
 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: