Merge branch 'main' into activation-ops

NVIDIA · Sep 26, 2024 · 6db1ddb · 6db1ddb
2 parents e1566e6 + 209b8e5
commit 6db1ddb
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git 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

diff --git 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"