github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/kernel/jit/__init__.py

11 lines
317 B
Python
Raw Normal View History

[misc] update pre-commit and run all files (#4752) * [misc] update pre-commit * [misc] run pre-commit * [misc] remove useless configuration files * [misc] ignore cuda for clang-format
2023-09-19 06:20:26 +00:00
from .bias_dropout_add import bias_dropout_add_fused_inference, bias_dropout_add_fused_train
refactor kernel (#142)
2022-01-13 08:47:17 +00:00
from .bias_gelu import bias_gelu_impl
[misc] update pre-commit and run all files (#4752) * [misc] update pre-commit * [misc] run pre-commit * [misc] remove useless configuration files * [misc] ignore cuda for clang-format
2023-09-19 06:20:26 +00:00
from .option import set_jit_fusion_options
add colossalai kernel module (#55)
2021-12-21 04:19:52 +00:00
refactor kernel (#142)
2022-01-13 08:47:17 +00:00
__all__ = [
[misc] update pre-commit and run all files (#4752) * [misc] update pre-commit * [misc] run pre-commit * [misc] remove useless configuration files * [misc] ignore cuda for clang-format
2023-09-19 06:20:26 +00:00
"bias_dropout_add_fused_train",
"bias_dropout_add_fused_inference",
"bias_gelu_impl",
"set_jit_fusion_options",
refactor kernel (#142)
2022-01-13 08:47:17 +00:00
]