Skip to content

Assembler for NVIDIA Volta and Turing GPUs

License

Notifications You must be signed in to change notification settings

HKBU-HPML/turingas

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

26 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

TuringAs

An open source SASS assembler for NVIDIA Volta and Turing GPUs.

Requirements:

  • Python >= 3.6

Supported hardware:

All NVIDIA Volta (SM70) and Turing (SM75) GPUs.

Other features:

  • Include files.
  • Inline python code.

Install the library

python setup.py install

Use the library

python -m turingas.main -i <input.sass> -o <output.cubin> -arch <arch>

# E.g.
python -m turingas.main -i input.sass -o output.cubin -arch 75 # 75 for Turing

Related projects

AsFermi, an SASS assembler for NVIDIA Fermi GPUs. By Hou Yunqing.

MaxAs, an SASS assembler for NVIDIA Maxwell and Pascal. By Scott Gray.

KeplerAs, an SASS assembler for NVIDIA Kepler. By Xiuxia Zhang.

TODO list

To support following instructions:

  • Type conversion instructions (I2I, I2F, F2I, F2F)
  • MUFU # Multifunction. E.g., sin, cos.
  • LDSM # Load matrix from shared memory.
  • Texture instructions
  • Surface instructions
  • Unified data path instructions
  • Other (ISCADD, CALL, JMP ...)

Citation

If you find this tool helpful, please cite:

@inproceedings{da2020winograd-conv,
  author    = {Da Yan and
               Wei Wang and
               Xiaowen Chu},
  title     = {Optimizing Batched Winograd Convolution on GPUs},
  booktitle = {25th ACM SIGPLAN Symposium on Principles and Practice of Parallel Programming (PPoPP '20)},
  year      = {2020},
  address   = {San Diego, CA, USA},
  publisher = {ACM},
}

This project is released under the MIT License.

-- Da Yan

About

Assembler for NVIDIA Volta and Turing GPUs

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 86.4%
  • Cuda 9.1%
  • CSS 3.9%
  • Shell 0.6%