disable torch-wheel

Author: liyuzhuo

.github/workflows/qa-l0-pytorch-wheel.yml

@@ -3,10 +3,10 @@ name: QA Pytorch Wheel
 on:
   push:
     branches:
-      - main
+      - __disabled_do_not_remove__
   pull_request:
     branches:
-      - main
+      - __disabled_do_not_remove__
 
 concurrency:
   group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}-${{ github.actor }}

qa/L0_pytorch_wheel/test.sh

@@ -27,6 +27,7 @@ VERSION=`cat $TE_PATH/build_tools/VERSION.txt`
 WHL_BASE="transformer_engine-${VERSION}"
 
 # Core wheel.
+rm -rf dist/*.whl 2>/dev/null || true  # Clean up any existing wheels
 NVTE_RELEASE_BUILD=1 pip3 wheel --no-build-isolation -vvv --wheel-dir ./dist . || error_exit "Failed to setup bdist_wheel"
 wheel unpack dist/${WHL_BASE}-* || error_exit "Failed to unpack dist/${WHL_BASE}-*.whl"
 sed -i "s/Name: transformer-engine/Name: transformer-engine-cu12/g" "transformer_engine-${VERSION}/transformer_engine-${VERSION}.dist-info/METADATA"
@@ -44,6 +45,8 @@ pip3 install --no-build-isolation --no-deps -vvv dist/* || error_exit "Failed to
 cd $TE_PATH
 pip3 install --no-build-isolation --no-deps -vvv dist/*.whl || error_exit "Failed to install dist/*.whl --no-deps"
 
+export TE_LIB_PATH=$(python -c "import site; print(site.getsitepackages()[0])")/transformer_engine
+
 python3 $TE_PATH/tests/pytorch/test_sanity_import.py || test_fail "test_sanity_import.py"
 
 if [ "$RET" -ne 0 ]; then