PyTorch: fix use of NVTXv3

fwyzard · fwyzard · commit 05125b249dd1 · 2025-05-16T13:16:26.000+02:00
diff --git a/pytorch-cluster.spec b/pytorch-cluster.spec
@@ -42,6 +42,7 @@ cmake ../%{n}-%{realversion} \
 %if 0%{!?without_cuda:1}
     -DWITH_CUDA=${USE_CUDA} \
     -DTORCH_CUDA_ARCH_LIST="%{cuda_arch_float}" \
+    -Dnvtx3_dir=${CUDA_ROOT}/include \
 %endif
     -DBUILD_SHARED_LIBS=ON
 
diff --git a/pytorch-scatter.spec b/pytorch-scatter.spec
@@ -43,6 +43,7 @@ cmake ../%{n}-%{realversion} \
 %if 0%{!?without_cuda:1}
     -DWITH_CUDA=${USE_CUDA} \
     -DTORCH_CUDA_ARCH_LIST="%{cuda_arch_float}" \
+    -Dnvtx3_dir=${CUDA_ROOT}/include \
 %endif
     -DBUILD_SHARED_LIBS=ON
 
diff --git a/pytorch-sparse.spec b/pytorch-sparse.spec
@@ -43,6 +43,7 @@ cmake ../%{n}-%{realversion} \
 %if 0%{!?without_cuda:1}
     -DWITH_CUDA=${USE_CUDA} \
     -DTORCH_CUDA_ARCH_LIST="%{cuda_arch_float}" \
+    -Dnvtx3_dir=${CUDA_ROOT}/include \
 %endif
     -DBUILD_SHARED_LIBS=ON
 
diff --git a/pytorch.spec b/pytorch.spec
@@ -45,6 +45,7 @@ cmake ../%{n}-%{realversion} \
 %if 0%{!?without_cuda:1}
     -DUSE_CUDA=${USE_CUDA} \
     -DTORCH_CUDA_ARCH_LIST="%{cuda_arch_float}" \
+    -Dnvtx3_dir=${CUDA_ROOT}/include \
     -DUSE_CUDNN=${USE_CUDA} \
     -DCUDNN_ROOT=${CUDNN_ROOT} \
 %else
