Skip to content

Commit 9740dca

Browse files
committed
only change im making is running on pushes to main instead of just new pins
1 parent e94ea4b commit 9740dca

File tree

1 file changed

+19
-7
lines changed

1 file changed

+19
-7
lines changed

.github/workflows/inductor.yml

Lines changed: 19 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
11
name: Test build/test linux gpu
22

33
on:
4-
pull_request:
5-
branches: [ main, "*"] #will remove this once we see everything is working fine
4+
push:
5+
branches: [ main ]
66
workflow_dispatch:
77
inputs:
88
triton_pin:
@@ -18,7 +18,6 @@ jobs:
1818
build-test:
1919
uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
2020
with:
21-
continue-on-error: true
2221
runner: linux.g5.48xlarge.nvidia.gpu
2322
gpu-arch-type: cuda
2423
gpu-arch-version: "12.1"
@@ -37,10 +36,24 @@ jobs:
3736
pip install ninja==1.11.1.1 cmake==3.30.2 wheel==0.44.0
3837
export llvm_hash=$(cat cmake/llvm-hash.txt)
3938
echo "llvm_hash: $llvm_hash"
40-
pushd python
41-
pip install .
42-
pushd ..
4339
pushd ..
40+
echo "Cloning llvm-project"
41+
git clone https://github.com/llvm/llvm-project.git
42+
pushd llvm-project
43+
echo "Checking out llvm hash"
44+
git checkout "$llvm_hash"
45+
mkdir build
46+
pushd build
47+
echo "Building llvm"
48+
cmake -G Ninja -DCMAKE_BUILD_TYPE=Release -DLLVM_ENABLE_ASSERTIONS=ON ../llvm -DLLVM_ENABLE_PROJECTS="mlir;llvm" -DLLVM_TARGETS_TO_BUILD="host;NVPTX;AMDGPU"
49+
ninja
50+
export LLVM_BUILD_DIR=$(pwd)
51+
popd
52+
popd
53+
popd
54+
LLVM_INCLUDE_DIRS=$LLVM_BUILD_DIR/include LLVM_LIBRARY_DIR=$LLVM_BUILD_DIR/lib LLVM_SYSPATH=$LLVM_BUILD_DIR pip install -e python
55+
echo "Installing triton python package"
56+
popd
4457
echo "Cloning pytorch"
4558
git clone https://github.com/pytorch/pytorch.git
4659
pushd pytorch
@@ -53,5 +66,4 @@ jobs:
5366
echo "Installing magma-cuda121"
5467
conda install -y -c pytorch magma-cuda121
5568
python setup.py install
56-
pip freeze
5769
pytest -n 1 test/inductor/test_torchinductor.py

0 commit comments

Comments
 (0)