Files
ROCm/scripts/amd/pytorch.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

9 lines
295 B
Bash
Executable File

# pip install transformers
# pip install --upgrade diffusers[torch]
# cd ../stuff/stable_diff
# python run.py
cd ../pytorch_rocm/
# TORCHINDUCTOR_COMPILE_THREADS=1 pytest test/inductor/test_torchinductor.py -k "test_views4_cuda"
pytest test/inductor/test_torchinductor.py -k "test_views4_cuda"