This website requires JavaScript.
Explore
Help
Register
Sign In
github
/
ColossalAI
Watch
1
Star
0
Fork
0
You've already forked ColossalAI
mirror of
https://github.com/hpcaitech/ColossalAI.git
synced
2026-04-26 01:35:21 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
Files
31fe84237b22d183a7de9db153f3b6fb739732f6
ColossalAI
/
colossalai
/
kernel
/
op_builder
History
Jiarui Fang
54de05da5d
[builder] polish builder with better base class (
#2216
)
...
* [builder] polish builder * remove print
2022-12-28 19:45:49 +08:00
..
__init__.py
[builder] multihead attn runtime building (
#2203
)
2022-12-27 16:06:09 +08:00
builder.py
[builder] polish builder with better base class (
#2216
)
2022-12-28 19:45:49 +08:00
cpu_adam.py
[builder] polish builder with better base class (
#2216
)
2022-12-28 19:45:49 +08:00
fused_optim.py
[builder] polish builder with better base class (
#2216
)
2022-12-28 19:45:49 +08:00
multi_head_attn.py
[builder] polish builder with better base class (
#2216
)
2022-12-28 19:45:49 +08:00
utils.py
[buider] use builder() for cpu adam and fused optim in setup.py (
#2187
)
2022-12-23 16:05:13 +08:00