Skip to content
This repository has been archived by the owner on Sep 27, 2024. It is now read-only.

Latest commit

 

History

History
13 lines (11 loc) · 569 Bytes

README.md

File metadata and controls

13 lines (11 loc) · 569 Bytes

Reproducer for IREE Bug

Installation

  • Create and activate a virtual environemt with python 3.10
  • Get your CUDA version with nvidia-smi
  • for CUDA 11 run pip install -r requirements-torch-cu118.txt
  • for CUDA 12 run pip install -r requirements-torch-cu121.txt
  • run pip install -r requirements-iree.txt
  • run pip install -r requirements-shark-turbine.txt
  • run ./compile.sh (this should give no output)
  • run ./run.sh (this should output EXEC @main several times)
  • run ./compares.sh (this should output Max difference: xxx several times)