Skip to content

Commit

Permalink
[Backend] Refactor codegen to separate GPU/CPU code generation (#176)
Browse files Browse the repository at this point in the history 
* refactor codegen to separate cpu and gpu backend

* add float16 and bfloat16 headers

* register bfloat16 primitive functions

* register fp16 primitive functions

* formatting

* modify codegen

* rename bfloat16 and float16 type

* fix rename

* fix comment
  • Loading branch information
@LDY1998
LDY1998 committed Apr 21, 2023
1 parent ed61049 commit f361211
Show file tree
Hide file tree
Showing 6 changed files with 1,998 additions and 134 deletions.

0 comments on commit f361211

Please sign in to comment.