ColossalAI/tests
Hongxin Liu 4f68b3f10c
[kernel] support pure fp16 for cpu adam and update gemini optim tests (#4921)
* [kernel] support pure fp16 for cpu adam (#4896)

* [kernel] fix cpu adam kernel for pure fp16 and update tests (#4919)

* [kernel] fix cpu adam

* [test] update gemini optim test
2023-10-16 21:56:53 +08:00
..
components_to_test
kit
test_analyzer
test_auto_parallel
test_autochunk
test_booster
test_checkpoint_io
test_cluster
test_config
test_device
test_fx
test_gptq
test_infer
test_infer_ops
test_lazy
test_legacy
test_moe
test_optimizer [kernel] support pure fp16 for cpu adam and update gemini optim tests (#4921) 2023-10-16 21:56:53 +08:00
test_pipeline
test_shardformer
test_smoothquant
test_tensor
test_utils
test_zero [kernel] support pure fp16 for cpu adam and update gemini optim tests (#4921) 2023-10-16 21:56:53 +08:00
__init__.py