ColossalAI/colossalai/kernel/cuda_native
ver217 12b4887097
[hotfix] fix CPUAdam kernel nullptr (#1410)
2022-08-05 19:45:45 +08:00
..
csrc [hotfix] fix CPUAdam kernel nullptr (#1410) 2022-08-05 19:45:45 +08:00
__init__.py
layer_norm.py
multihead_attention.py
scaled_softmax.py