kernels of different DL funcs
- ELU (fp32, fp16, fp16x2, fp16x8_packed)
- GeLU (fp32, fp16, fp16x4_packed)
- Sigmoid (fp32, fp16, fp16x8_packed)
- ReLU (fp32, fp16)
- Swish (fp32, fp16)
- similar kernel to
torch.nn.functional.embeddingin fp32 & fp16
-
Notifications
You must be signed in to change notification settings - Fork 1
SwekeR-463/kernels
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
learning & making kernels in cuda / triton
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published