From a44cb72439574973809449e596464174e5a6fe37 Mon Sep 17 00:00:00 2001 From: Kirthi Shankar Sivamani Date: Tue, 24 Sep 2024 11:59:19 -0700 Subject: [PATCH 1/2] Update list of CI users (#1203) Add new users to CI Signed-off-by: Kirthi Shankar Sivamani --- .github/workflows/trigger-ci.yml | 2 ++ 1 file changed, 2 insertions(+) diff --git a/.github/workflows/trigger-ci.yml b/.github/workflows/trigger-ci.yml index 711518d6fe..c7a85029e6 100644 --- a/.github/workflows/trigger-ci.yml +++ b/.github/workflows/trigger-ci.yml @@ -37,6 +37,8 @@ jobs: || github.actor == 'huanghua1994' || github.actor == 'mgoldfarb-nvidia' || github.actor == 'pggPL' + || github.actor == 'vasunvidia' + || github.actor == 'erhoo82' ) steps: - name: Check if comment is issued by authorized person From 209b8e5a801a7747adfa42235aedeb6bfd805874 Mon Sep 17 00:00:00 2001 From: Sangkug Lym Date: Tue, 24 Sep 2024 21:24:28 -0700 Subject: [PATCH 2/2] fix NVTE_UB_WITH_MPI read (#1194) * fix NVTE_UB_WITH_MPI read Signed-off-by: Sangkug Lym * Add default value Signed-off-by: Sangkug Lym --------- Signed-off-by: Sangkug Lym Co-authored-by: Kirthi Shankar Sivamani --- build_tools/pytorch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/build_tools/pytorch.py b/build_tools/pytorch.py index 7aff2db324..9152229d2f 100644 --- a/build_tools/pytorch.py +++ b/build_tools/pytorch.py @@ -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"