Skip to content
forked from ROCm/MISA

Code generator for implicit gemm algorithm

License

Notifications You must be signed in to change notification settings

qianfengz/iGEMMgen

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

94 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

under rapid development

iGEMMgen

Code generator for implicit gemm algorithm (generic tensor contraction)

Generate kernel

since we use f-string of python, we require python >= 3.6 to run.

# generate code based on tunable configuration
python3 igemm_codegen.py config/igemm_bwd_gtc.config

The output file will result in out directory. result in a assembly file *.s, a codeobject *.hsaco and a host driver executable conv_driver.exe. This executable accept same cmdline argument as MIOpenDriver. e.g.

./conv_driver.exe  conv -n 128 -c 1024 -H 17 -W 17 -k 1024  -y 1 -x 1 -p 0 -q 0 -u 1 -v 1 -l 1 -j 1 -F 2

more description to be added

About

Code generator for implicit gemm algorithm

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 75.3%
  • C++ 24.2%
  • Shell 0.5%