├── .clang-format ├── .gitignore ├── CMakeLists.txt ├── README.md ├── include ├── dsp │ ├── dma_utils.h │ ├── hmx_mgr.h │ ├── hmx_utils.h │ ├── hvx_convert.h │ ├── hvx_internal.h │ ├── hvx_math.h │ ├── mmap_mgr.h │ ├── op_executor.h │ ├── ops.h │ ├── power.h │ ├── quants.h │ ├── utils.h │ ├── vtcm_mgr.h │ └── worker_pool.h ├── host │ ├── op_export.h │ └── session.h ├── htp_ops.idl ├── message.h └── op_reg.h └── src ├── dsp ├── commu.c ├── hmx_mgr.c ├── mmap_mgr.cc ├── op_executor.cc ├── op_tests.cc ├── ops │ ├── flash_attn.c │ ├── flash_attn_sp_hdim.c │ ├── mat_mul.c │ ├── mm_benchmark.c │ ├── precompute_table.c │ └── rms_norm.c ├── power.c ├── vtcm_mgr.cc └── worker_pool.c └── host ├── op_export.c ├── session.c └── test.c /.clang-format: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/.clang-format -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- 1 | # build output 2 | android* 3 | hexagon* 4 | 5 | -------------------------------------------------------------------------------- /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/README.md -------------------------------------------------------------------------------- /include/dsp/dma_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/include/dsp/dma_utils.h -------------------------------------------------------------------------------- /include/dsp/hmx_mgr.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/include/dsp/hmx_mgr.h -------------------------------------------------------------------------------- /include/dsp/hmx_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/include/dsp/hmx_utils.h -------------------------------------------------------------------------------- /include/dsp/hvx_convert.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/include/dsp/hvx_convert.h -------------------------------------------------------------------------------- /include/dsp/hvx_internal.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/include/dsp/hvx_internal.h -------------------------------------------------------------------------------- /include/dsp/hvx_math.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/include/dsp/hvx_math.h -------------------------------------------------------------------------------- /include/dsp/mmap_mgr.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/include/dsp/mmap_mgr.h -------------------------------------------------------------------------------- /include/dsp/op_executor.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/include/dsp/op_executor.h -------------------------------------------------------------------------------- /include/dsp/ops.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/include/dsp/ops.h -------------------------------------------------------------------------------- /include/dsp/power.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/include/dsp/power.h -------------------------------------------------------------------------------- /include/dsp/quants.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/include/dsp/quants.h -------------------------------------------------------------------------------- /include/dsp/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/include/dsp/utils.h -------------------------------------------------------------------------------- /include/dsp/vtcm_mgr.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/include/dsp/vtcm_mgr.h -------------------------------------------------------------------------------- /include/dsp/worker_pool.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/include/dsp/worker_pool.h -------------------------------------------------------------------------------- /include/host/op_export.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/include/host/op_export.h -------------------------------------------------------------------------------- /include/host/session.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/include/host/session.h -------------------------------------------------------------------------------- /include/htp_ops.idl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/include/htp_ops.idl -------------------------------------------------------------------------------- /include/message.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/include/message.h -------------------------------------------------------------------------------- /include/op_reg.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/include/op_reg.h -------------------------------------------------------------------------------- /src/dsp/commu.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/src/dsp/commu.c -------------------------------------------------------------------------------- /src/dsp/hmx_mgr.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/src/dsp/hmx_mgr.c -------------------------------------------------------------------------------- /src/dsp/mmap_mgr.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/src/dsp/mmap_mgr.cc -------------------------------------------------------------------------------- /src/dsp/op_executor.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/src/dsp/op_executor.cc -------------------------------------------------------------------------------- /src/dsp/op_tests.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/src/dsp/op_tests.cc -------------------------------------------------------------------------------- /src/dsp/ops/flash_attn.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/src/dsp/ops/flash_attn.c -------------------------------------------------------------------------------- /src/dsp/ops/flash_attn_sp_hdim.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/src/dsp/ops/flash_attn_sp_hdim.c -------------------------------------------------------------------------------- /src/dsp/ops/mat_mul.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/src/dsp/ops/mat_mul.c -------------------------------------------------------------------------------- /src/dsp/ops/mm_benchmark.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/src/dsp/ops/mm_benchmark.c -------------------------------------------------------------------------------- /src/dsp/ops/precompute_table.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/src/dsp/ops/precompute_table.c -------------------------------------------------------------------------------- /src/dsp/ops/rms_norm.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/src/dsp/ops/rms_norm.c -------------------------------------------------------------------------------- /src/dsp/power.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/src/dsp/power.c -------------------------------------------------------------------------------- /src/dsp/vtcm_mgr.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/src/dsp/vtcm_mgr.cc -------------------------------------------------------------------------------- /src/dsp/worker_pool.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/src/dsp/worker_pool.c -------------------------------------------------------------------------------- /src/host/op_export.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/src/host/op_export.c -------------------------------------------------------------------------------- /src/host/session.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/src/host/session.c -------------------------------------------------------------------------------- /src/host/test.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haozixu/htp-ops-lib/HEAD/src/host/test.c --------------------------------------------------------------------------------