File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -121,28 +121,23 @@ jobs:
121121 - name : Environment Setup
122122 if : steps.should_run.outputs.should_run == 'true'
123123 env :
124-
125124 NVTE_WITH_MACA : ' 1'
126125 NVTE_WITH_CUDA : ' 0'
126+ MACA_HOME : /opt/maca
127127 NVCC : /opt/maca/bin/mcc
128128 CUDA_HOME : /opt/maca
129-
130- PATH : /opt/maca/bin:${{ env.PATH }}
131- LD_LIBRARY_PATH : /opt/maca/lib:${{ env.LD_LIBRARY_PATH }}
132129 run : |
130+ export PATH=/opt/maca/bin:$PATH
131+ export LD_LIBRARY_PATH=/opt/maca/lib:/usr/local/maca/lib:${LD_LIBRARY_PATH:-}
132+
133133 set -euo pipefail
134134 # 1. Execute hardware-specific setup (e.g. conda activate, export paths)
135135
136136 #! setup_commands from all_tests_xxx.yml(like all_tests_metax.yml)
137137 ${{ inputs.setup_commands }}
138138
139- export PATH=/opt/maca/bin:$PATH
140- export LD_LIBRARY_PATH=/opt/maca/lib:/usr/local/maca/lib:$LD_LIBRARY_PATH
141- export MACA_HOME=/opt/maca
142- export NVCC=/opt/maca/bin/mcc
143- export NVTE_WITH_MACA=1
144- export NVTE_WITH_CUDA=0
145139
140+
146141 which python3
147142 python3 -c "import torch; print(f'Native Torch Version: {torch.__version__}')"
148143
You can’t perform that action at this time.
0 commit comments