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

lucas-camp/torch_iree_cuda_repr

BranchesTags

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
.gitignore
.gitignore
 
 
README.md
README.md
 
 
compare.py
compare.py
 
 
compare_torch_mlir.sh
compare_torch_mlir.sh
 
 
compare_turbine.sh
compare_turbine.sh
 
 
compile_torch_mlir.sh
compile_torch_mlir.sh
 
 
compile_turbine.sh
compile_turbine.sh
 
 
generate_mlir_module_torch_mlir.py
generate_mlir_module_torch_mlir.py
 
 
generate_mlir_module_turbine.py
generate_mlir_module_turbine.py
 
 
requirements-iree.txt
requirements-iree.txt
 
 
requirements-shark-turbine.txt
requirements-shark-turbine.txt
 
 
requirements-torch-cpu.txt
requirements-torch-cpu.txt
 
 
requirements-torch-cu118.txt
requirements-torch-cu118.txt
 
 
requirements-torch-cu121.txt
requirements-torch-cu121.txt
 
 
requirements-torch-mlir-torch.txt
requirements-torch-mlir-torch.txt
 
 
requirements-torch-mlir.txt
requirements-torch-mlir.txt
 
 
run_torch_mlir.sh
run_torch_mlir.sh
 
 
run_turbine.sh
run_turbine.sh
 
 

Repository files navigation

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)

About

No description, website, or topics provided.

Resources

Readme
Activity

Stars

1 star

Watchers

1 watching

Forks

0 forks
Report repository

Releases

No releases published

Packages

No packages published

Languages