Skip to content

Commit

Permalink
fix NVTE_UB_WITH_MPI read (#1194)
Browse files Browse the repository at this point in the history
* fix NVTE_UB_WITH_MPI read

Signed-off-by: Sangkug Lym <slym@nvidia.com>

* Add default value

Signed-off-by: Sangkug Lym <slym@nvidia.com>

---------

Signed-off-by: Sangkug Lym <slym@nvidia.com>
Co-authored-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
  • Loading branch information
erhoo82 and ksivaman authored Sep 25, 2024
1 parent a44cb72 commit 209b8e5
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion build_tools/pytorch.py
Original file line number Diff line number Diff line change
Expand Up @@ -88,7 +88,7 @@ def setup_pytorch_extension(
# Libraries
library_dirs = []
libraries = []
if os.getenv("NVTE_UB_WITH_MPI"):
if bool(int(os.getenv("NVTE_UB_WITH_MPI", 0))):
assert (
os.getenv("MPI_HOME") is not None
), "MPI_HOME must be set when compiling with NVTE_UB_WITH_MPI=1"
Expand Down

0 comments on commit 209b8e5

Please sign in to comment.