Skip to content

Ascend/triton-ascend-examples

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

88 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Triton Ascend Example

News

About

Since the micro architecture of AI chips are different from different vendors, so is Ascend, which is quite different from NV GPU. This repo provides the best practices of writing computing kernel using Triton Language on Ascend. To show the best practices, here provide three types of example:

  • basic: basic examples
  • op_extension: examples of extended operations provided by Triton Ascend
  • transformer: examples used in transformer models and inference engine for LLM

How run it

Prerequisite

  • CANN
  • Pytorch and Pytorch Ascend
  • Triton Ascend

License

Follow MIT license, check LICENSE file

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 9

Languages