From 435ca68f15181275411346e02d7b00035b071f23 Mon Sep 17 00:00:00 2001 From: fzyzcjy Date: Sat, 5 Jul 2025 14:00:03 +0800 Subject: [PATCH 1/2] more --- setup.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/setup.py b/setup.py index 287ea9ba..4d409782 100644 --- a/setup.py +++ b/setup.py @@ -21,6 +21,8 @@ nvcc_dlink = [] extra_link_args = [] + nvcc_flags += os.environ.get("DEEPEP_EXTRA_NVCC_FLAGS", "").split(" ") + # NVSHMEM flags if disable_nvshmem: cxx_flags.append('-DDISABLE_NVSHMEM') From b677bac6e245c444196bcff7d939f2a07b04d37a Mon Sep 17 00:00:00 2001 From: fzyzcjy Date: Sat, 5 Jul 2025 14:02:13 +0800 Subject: [PATCH 2/2] more --- setup.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 4d409782..bf9bf412 100644 --- a/setup.py +++ b/setup.py @@ -21,7 +21,8 @@ nvcc_dlink = [] extra_link_args = [] - nvcc_flags += os.environ.get("DEEPEP_EXTRA_NVCC_FLAGS", "").split(" ") + if (extra_nvcc_flags := os.environ.get("DEEPEP_EXTRA_NVCC_FLAGS")) is not None: + nvcc_flags += extra_nvcc_flags.split(" ") # NVSHMEM flags if disable_nvshmem: