mirror of https://github.com/hpcaitech/ColossalAI
279300dc5f
* refactor compilation mechanism and unified multi hw * fix file path bug * add init.py to make pybind a module to avoid relative path error caused by softlink * delete duplicated micros * fix micros bug in gcc |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| scaled_masked_softmax.cpp | ||
| scaled_masked_softmax_cuda.py | ||
| scaled_upper_triang_masked_softmax.cpp | ||
| scaled_upper_triangle_masked_softmax_cuda.py | ||