Interger softmax function with different bitwidth for Associate Processor.
Just replace softmax fucntion with i_softmax_wrapper in each transformer codes.
ython test_gpu_softmax.py
-
Notifications
You must be signed in to change notification settings - Fork 0
Kimho666/softmap
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Interger softmax function with different bitwidth for Associate Processor.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published