Skip to content

GitLab

Unverified Commit f0295f9d authored by Phuong Nguyen's avatar Phuong Nguyen Committed by GitHub
Browse files

CMake to respect MAX_JOBS or NVTE_MAX_JOBS (#2319) 



fix max jobs
Signed-off-by: default avatarPhuong Nguyen <phuonguyen@nvidia.com>
parent c6cbcc85
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
transformer_engine/common/CMakeLists.txt
View file @ f0295f9d
......@@ -352,10 +352,10 @@ set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} --expt-relaxed-constexpr")
set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -O3")
# Number of parallel build jobs
if(ENV{MAX_JOBS})
set(BUILD_JOBS_STR "$ENV{MAX_JOBS}")
elseif(ENV{NVTE_BUILD_MAX_JOBS})
set(BUILD_JOBS_STR "$ENV{NVTE_BUILD_MAX_JOBS}")
if($ENV{MAX_JOBS})
set(BUILD_JOBS_STR $ENV{MAX_JOBS})
elseif($ENV{NVTE_BUILD_MAX_JOBS})
set(BUILD_JOBS_STR $ENV{NVTE_BUILD_MAX_JOBS})
else()
set(BUILD_JOBS_STR "max")
endif()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment