Adding some required environment varaibles.

sandeepd-nv · sandeepd-nv · commit 5b982299168e · 2024-08-07T09:53:55.000-07:00
diff --git a/continuous_integration/scripts/build b/continuous_integration/scripts/build
@@ -5,6 +5,9 @@ build_ci() {
 
     cd "${REPO_DIR}"
 
+    export CUDA_HOME="${CONDA_PREFIX}/targets/x86_64-linux"
+    export PARALLEL_LEVEL=$(nproc --ignore 1)
+
     python setup.py build_ext --inplace
 }
 

Original file line number	Diff line number	Diff line change
`@@ -5,6 +5,9 @@ build_ci() {`
`5`	`5`
`6`	`6`	`cd "${REPO_DIR}"`
`7`	`7`
	`8`	`+ export CUDA_HOME="${CONDA_PREFIX}/targets/x86_64-linux"`
	`9`	`+ export PARALLEL_LEVEL=$(nproc --ignore 1)`
	`10`	`+`
`8`	`11`	`python setup.py build_ext --inplace`
`9`	`12`	`}`
`10`	`13`