File tree Expand file tree Collapse file tree 1 file changed +19
-7
lines changed Expand file tree Collapse file tree 1 file changed +19
-7
lines changed Original file line number Diff line number Diff line change 1
1
name : Test build/test linux gpu
2
2
3
3
on :
4
- pull_request :
5
- branches : [ main, "*"] # will remove this once we see everything is working fine
4
+ push :
5
+ branches : [ main ]
6
6
workflow_dispatch :
7
7
inputs :
8
8
triton_pin :
18
18
build-test :
19
19
uses : pytorch/test-infra/.github/workflows/linux_job.yml@main
20
20
with :
21
- continue-on-error : true
22
21
runner : linux.g5.48xlarge.nvidia.gpu
23
22
gpu-arch-type : cuda
24
23
gpu-arch-version : " 12.1"
@@ -37,10 +36,24 @@ jobs:
37
36
pip install ninja==1.11.1.1 cmake==3.30.2 wheel==0.44.0
38
37
export llvm_hash=$(cat cmake/llvm-hash.txt)
39
38
echo "llvm_hash: $llvm_hash"
40
- pushd python
41
- pip install .
42
- pushd ..
43
39
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
44
57
echo "Cloning pytorch"
45
58
git clone https://github.com/pytorch/pytorch.git
46
59
pushd pytorch
53
66
echo "Installing magma-cuda121"
54
67
conda install -y -c pytorch magma-cuda121
55
68
python setup.py install
56
- pip freeze
57
69
pytest -n 1 test/inductor/test_torchinductor.py
You can’t perform that action at this time.
0 commit comments