傅剑寒 279300dc5f [Inference/Refactor] Refactor compilation mechanism and unified multi hw (#5613) ...

* 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

2024-04-24 14:17:54 +08:00

..

__init__.py

[Inference/Refactor] Refactor compilation mechanism and unified multi hw (#5613)

2024-04-24 14:17:54 +08:00

scaled_masked_softmax_cuda.py

[Inference/Refactor] Refactor compilation mechanism and unified multi hw (#5613)

2024-04-24 14:17:54 +08:00

scaled_masked_softmax.cpp

[Inference/Refactor] Refactor compilation mechanism and unified multi hw (#5613)

2024-04-24 14:17:54 +08:00

scaled_upper_triang_masked_softmax.cpp

[Inference/Refactor] Refactor compilation mechanism and unified multi hw (#5613)

2024-04-24 14:17:54 +08:00

scaled_upper_triangle_masked_softmax_cuda.py

[Inference/Refactor] Refactor compilation mechanism and unified multi hw (#5613)

2024-04-24 14:17:54 +08:00