Files
ROCm/scripts/amd/build.sh
Michael Melesse 9cc141b12d assume ROCM device
This is a combination of 7 commits.

use pyt nightly with root

repro with pytorch unit test

hardcode isROCM to true

set is_cuda to False

ignore cc arg

clean up

match triton-mlir branch
2023-05-04 16:46:59 -05:00

16 lines
261 B
Bash
Executable File

set -x
cd python
pip uninstall -y triton
bash scripts/amd/clean.sh
export MLIR_ENABLE_DUMP=1
export LLVM_IR_ENABLE_DUMP=1
export AMDGCN_ENABLE_DUMP=1
export TRITON_USE_ROCM=ON
# pip install -U matplotlib pandas filelock tabulate
pip install --verbose -e .