InfiniTensor/include/bang
zhangyunze 67b2bcb7d5
fix mlu some kernel registration & gather op (#210)
* fix: fix bang build/kernel registration | test_onnx

* delete assert float

* fix gather

* fix CMakeLists and Reshape

* fix cncl ops

* add hardsigmoid/hardswish

* fix

* add invalid datatype exception

* fix gather

* fix gather indices type

* fix gather/prelu/hardsigmoid on mlu

* fix format

* fix

---------

Co-authored-by: Bolun Zhang <48948016+Chamberlain0w0@users.noreply.github.com>
Co-authored-by: Haojie Wang <haojie0429@gmail.com>
Co-authored-by: Zhang Bolun <Chamberlain0w0@gmail.com>
2024-02-01 15:02:02 +08:00
..
bang_common.h fix mlu some kernel registration & gather op (#210) 2024-02-01 15:02:02 +08:00
bang_kernel_without_config.h Add bangc runtime and element-wise kernels 2022-09-22 16:57:39 +08:00
bang_runtime.h Bang cncl (#163) 2024-01-03 13:28:03 +08:00
cncl_communicator.h Bang cncl (#163) 2024-01-03 13:28:03 +08:00
operator_timer.h Support perf bang 1115 (#57) 2023-03-29 13:52:56 +08:00