├── .clang-format ├── .github └── workflows │ └── main.yaml ├── .gitignore ├── README.md ├── include ├── data_type.h ├── device.h ├── export.h ├── handle.h ├── handle │ └── handle_export.h ├── infini_operators.h ├── operators.h ├── ops │ ├── add │ │ └── add.h │ ├── attention │ │ └── attention.h │ ├── avg_pool │ │ └── avg_pool.h │ ├── causal_softmax │ │ └── causal_softmax.h │ ├── conv │ │ └── conv.h │ ├── expand │ │ └── expand.h │ ├── gemm │ │ └── gemm.h │ ├── global_avg_pool │ │ └── global_avg_pool.h │ ├── matmul │ │ └── matmul.h │ ├── max_pool │ │ └── max_pool.h │ ├── mlp │ │ └── mlp.h │ ├── random_sample │ │ └── random_sample.h │ ├── rearrange │ │ └── rearrange.h │ ├── relu │ │ └── relu.h │ ├── rms_norm │ │ └── rms_norm.h │ ├── rotary_embedding │ │ └── rotary_embedding.h │ └── swiglu │ │ └── swiglu.h ├── status.h ├── tensor.h └── tensor │ └── tensor_descriptor.h ├── operatorspy ├── __init__.py ├── data_layout.py ├── devices.py ├── liboperators.py ├── tests │ ├── __init__.py │ ├── add.py │ ├── attention.py │ ├── avg_pool.py │ ├── causal_softmax.py │ ├── conv.py │ ├── expand.py │ ├── gemm.py │ ├── global_avg_pool.py │ ├── matmul.py │ ├── max_pool.py │ ├── mlp.py │ ├── random_sample.py │ ├── rearrange.py │ ├── relu.py │ ├── rms_norm.py │ ├── rotary_embedding.py │ ├── swiglu.py │ └── test_utils.py └── utils.py ├── src ├── devices │ ├── ascend │ │ ├── CMakeLists.txt │ │ ├── Makefile │ │ ├── ascend_handle.cc │ │ ├── ascend_handle.h │ │ ├── common_ascend.cc │ │ ├── common_ascend.h │ │ ├── tensor_aclnn.cc │ │ └── tensor_aclnn.h │ ├── bang │ │ ├── bang_handle.cc │ │ ├── bang_handle.h │ │ └── common_bang.h │ ├── cpu │ │ ├── common_cpu.cc │ │ ├── common_cpu.h │ │ ├── cpu_handle.cc │ │ └── cpu_handle.h │ ├── cuda │ │ ├── common_cuda.h │ │ ├── cuda_handle.cc │ │ └── cuda_handle.h │ ├── handle.cc │ ├── maca │ │ ├── common_maca.h │ │ ├── maca_handle.cc │ │ └── maca_handle.h │ ├── musa │ │ ├── common_musa.h │ │ ├── musa_handle.cc │ │ ├── musa_handle.h │ │ └── pool.h │ └── pool.h ├── ops │ ├── add │ │ ├── cpu │ │ │ ├── add_cpu.cc │ │ │ └── add_cpu.h │ │ ├── cuda │ │ │ ├── add.cc │ │ │ ├── add.cu │ │ │ └── add.cuh │ │ ├── musa │ │ │ ├── add_musa.cc │ │ │ ├── add_musa.h │ │ │ └── add_musa.mu │ │ └── operator.cc │ ├── attention │ │ └── operator.cc │ ├── avg_pool │ │ └── operator.cc │ ├── causal_softmax │ │ ├── ascend │ │ │ ├── causal_softmax_aclnn.cc │ │ │ └── causal_softmax_aclnn.h │ │ ├── bang │ │ │ ├── causal_softmax_bang.cc │ │ │ ├── causal_softmax_bang.h │ │ │ ├── causal_softmax_bang.mlu │ │ │ ├── causal_softmax_cnnl.cc │ │ │ └── causal_softmax_cnnl.h │ │ ├── cpu │ │ │ ├── causal_softmax_cpu.cc │ │ │ └── causal_softmax_cpu.h │ │ ├── cuda │ │ │ ├── causal_softmax.cc │ │ │ ├── causal_softmax.cu │ │ │ └── causal_softmax.cuh │ │ ├── maca │ │ │ ├── causal_softmax_maca.cc │ │ │ ├── causal_softmax_maca.h │ │ │ └── causal_softmax_maca.maca │ │ ├── musa │ │ │ ├── causal_softmax_musa.cc │ │ │ ├── causal_softmax_musa.h │ │ │ └── causal_softmax_musa.mu │ │ └── operator.cc │ ├── conv │ │ ├── cpu │ │ │ ├── conv_cpu.cc │ │ │ └── conv_cpu.h │ │ ├── cuda │ │ │ ├── conv.cc │ │ │ ├── conv.cu │ │ │ └── conv.cuh │ │ └── operator.cc │ ├── expand │ │ ├── cpu │ │ │ ├── expand_cpu.cc │ │ │ └── expand_cpu.h │ │ ├── cuda │ │ │ ├── expand.cc │ │ │ ├── expand.cu │ │ │ └── expand.cuh │ │ ├── musa │ │ │ ├── expand_musa.cc │ │ │ ├── expand_musa.h │ │ │ └── expand_musa.mu │ │ └── operator.cc │ ├── gemm │ │ └── operator.cc │ ├── global_avg_pool │ │ ├── cpu │ │ │ ├── global_avg_pool_cpu.cc │ │ │ └── global_avg_pool_cpu.h │ │ ├── cuda │ │ │ ├── global_avg_pool.cc │ │ │ ├── global_avg_pool.cu │ │ │ └── global_avg_pool.cuh │ │ └── operator.cc │ ├── matmul │ │ ├── ascend │ │ │ ├── matmul_aclnn.cc │ │ │ └── matmul_aclnn.h │ │ ├── bang │ │ │ ├── matmul_cnnl.cc │ │ │ └── matmul_cnnl.h │ │ ├── blas.h │ │ ├── cpu │ │ │ ├── matmul_cpu.cc │ │ │ └── matmul_cpu.h │ │ ├── cuda │ │ │ ├── matmul_cuda.cc │ │ │ ├── matmul_cuda.cu │ │ │ └── matmul_cuda.h │ │ ├── maca │ │ │ ├── matmul_maca.cc │ │ │ ├── matmul_maca.h │ │ │ └── matmul_maca.maca │ │ ├── musa │ │ │ ├── matmul_musa.cc │ │ │ ├── matmul_musa.h │ │ │ └── matmul_musa.mu │ │ └── operator.cc │ ├── max_pool │ │ └── operator.cc │ ├── mlp │ │ └── operator.cc │ ├── pooling │ │ ├── cpu │ │ │ ├── pooling_cpu.cc │ │ │ └── pooling_cpu.h │ │ ├── cuda │ │ │ ├── pooling.cc │ │ │ ├── pooling.cu │ │ │ └── pooling.cuh │ │ ├── operator.cc │ │ └── pooling.h │ ├── random_sample │ │ ├── ascend │ │ │ ├── random_sample.cc │ │ │ ├── random_sample.h │ │ │ └── random_sample_kernel.cpp │ │ ├── bang │ │ │ ├── random_sample_bang.cc │ │ │ ├── random_sample_bang.h │ │ │ └── random_sample_bang.mlu │ │ ├── cpu │ │ │ ├── random_sample.cc │ │ │ └── random_sample_cpu.h │ │ ├── cuda │ │ │ ├── random_sample.cu │ │ │ ├── random_sample.cuh │ │ │ └── random_sample_cuda.cc │ │ ├── maca │ │ │ ├── random_sample_maca.cc │ │ │ ├── random_sample_maca.h │ │ │ └── random_sample_maca.maca │ │ ├── musa │ │ │ ├── random_sample_musa.cc │ │ │ ├── random_sample_musa.h │ │ │ └── random_sample_musa.mu │ │ └── operator.cc │ ├── rearrange │ │ ├── ascend │ │ │ ├── rearrange_aclnn.cc │ │ │ └── rearrange_aclnn.h │ │ ├── bang │ │ │ ├── rearrange_bang.cc │ │ │ ├── rearrange_bang.h │ │ │ └── rearrange_bang.mlu │ │ ├── cpu │ │ │ ├── rearrange_cpu.cc │ │ │ └── rearrange_cpu.h │ │ ├── cuda │ │ │ ├── rearrange.cc │ │ │ ├── rearrange.cu │ │ │ └── rearrange.cuh │ │ ├── maca │ │ │ ├── rearrange_maca.cc │ │ │ ├── rearrange_maca.h │ │ │ └── rearrange_maca.maca │ │ ├── musa │ │ │ ├── rearrange_musa.cc │ │ │ ├── rearrange_musa.h │ │ │ └── rearrange_musa.mu │ │ └── operator.cc │ ├── relu │ │ ├── cpu │ │ │ ├── relu_cpu.cc │ │ │ └── relu_cpu.h │ │ ├── cuda │ │ │ ├── relu.cc │ │ │ ├── relu.cu │ │ │ └── relu.cuh │ │ ├── musa │ │ │ ├── relu_musa.cc │ │ │ ├── relu_musa.h │ │ │ └── relu_musa.mu │ │ └── operator.cc │ ├── rms_norm │ │ ├── ascend │ │ │ ├── rms_norm_aclnn.cc │ │ │ └── rms_norm_aclnn.h │ │ ├── bang │ │ │ ├── rms_norm_bang.cc │ │ │ ├── rms_norm_bang.h │ │ │ └── rms_norm_bang.mlu │ │ ├── cpu │ │ │ ├── rms_norm_cpu.cc │ │ │ └── rms_norm_cpu.h │ │ ├── cuda │ │ │ ├── rms_norm.cc │ │ │ ├── rms_norm.cu │ │ │ └── rms_norm.cuh │ │ ├── maca │ │ │ ├── rms_norm_maca.cc │ │ │ ├── rms_norm_maca.h │ │ │ └── rms_norm_maca.maca │ │ ├── musa │ │ │ ├── rms_norm_musa.cc │ │ │ ├── rms_norm_musa.h │ │ │ └── rms_norm_musa.mu │ │ └── operator.cc │ ├── rotary_embedding │ │ ├── ascend │ │ │ ├── rotary_embedding.cc │ │ │ ├── rotary_embedding.h │ │ │ └── rotary_embedding_kernel.cpp │ │ ├── bang │ │ │ ├── rotary_embedding_bang.cc │ │ │ ├── rotary_embedding_bang.h │ │ │ └── rotary_embedding_bang.mlu │ │ ├── cpu │ │ │ ├── rotary_embedding_cpu.cc │ │ │ └── rotary_embedding_cpu.h │ │ ├── cuda │ │ │ ├── rotary_embedding.cc │ │ │ ├── rotary_embedding.cu │ │ │ └── rotary_embedding.cuh │ │ ├── maca │ │ │ ├── rotary_embedding_maca.cc │ │ │ ├── rotary_embedding_maca.h │ │ │ └── rotary_embedding_maca.maca │ │ ├── musa │ │ │ ├── rotary_embedding_musa.cc │ │ │ ├── rotary_embedding_musa.h │ │ │ └── rotary_embedding_musa.mu │ │ └── operator.cc │ ├── swiglu │ │ ├── ascend │ │ │ ├── swiglu.cc │ │ │ ├── swiglu.h │ │ │ └── swiglu_kernel.cpp │ │ ├── bang │ │ │ ├── swiglu_bang.cc │ │ │ ├── swiglu_bang.h │ │ │ └── swiglu_bang.mlu │ │ ├── cpu │ │ │ ├── swiglu_cpu.cc │ │ │ └── swiglu_cpu.h │ │ ├── cuda │ │ │ ├── swiglu.cu │ │ │ ├── swiglu.cuh │ │ │ └── swiglu_cuda.cc │ │ ├── maca │ │ │ ├── swiglu_maca.cc │ │ │ ├── swiglu_maca.h │ │ │ └── swiglu_maca.maca │ │ ├── musa │ │ │ ├── swiglu.mu │ │ │ ├── swiglu_musa.cc │ │ │ └── swiglu_musa.h │ │ └── operator.cc │ └── utils.h └── tensor │ └── tensor_descriptor.cc └── xmake.lua /.clang-format: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/.clang-format -------------------------------------------------------------------------------- /.github/workflows/main.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/.github/workflows/main.yaml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/README.md -------------------------------------------------------------------------------- /include/data_type.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/include/data_type.h -------------------------------------------------------------------------------- /include/device.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/include/device.h -------------------------------------------------------------------------------- /include/export.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/include/export.h -------------------------------------------------------------------------------- /include/handle.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/include/handle.h -------------------------------------------------------------------------------- /include/handle/handle_export.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/include/handle/handle_export.h -------------------------------------------------------------------------------- /include/infini_operators.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/include/infini_operators.h -------------------------------------------------------------------------------- /include/operators.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/include/operators.h -------------------------------------------------------------------------------- /include/ops/add/add.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/include/ops/add/add.h -------------------------------------------------------------------------------- /include/ops/attention/attention.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/include/ops/attention/attention.h -------------------------------------------------------------------------------- /include/ops/avg_pool/avg_pool.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/include/ops/avg_pool/avg_pool.h -------------------------------------------------------------------------------- /include/ops/causal_softmax/causal_softmax.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/include/ops/causal_softmax/causal_softmax.h -------------------------------------------------------------------------------- /include/ops/conv/conv.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/include/ops/conv/conv.h -------------------------------------------------------------------------------- /include/ops/expand/expand.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/include/ops/expand/expand.h -------------------------------------------------------------------------------- /include/ops/gemm/gemm.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/include/ops/gemm/gemm.h -------------------------------------------------------------------------------- /include/ops/global_avg_pool/global_avg_pool.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/include/ops/global_avg_pool/global_avg_pool.h -------------------------------------------------------------------------------- /include/ops/matmul/matmul.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/include/ops/matmul/matmul.h -------------------------------------------------------------------------------- /include/ops/max_pool/max_pool.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/include/ops/max_pool/max_pool.h -------------------------------------------------------------------------------- /include/ops/mlp/mlp.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/include/ops/mlp/mlp.h -------------------------------------------------------------------------------- /include/ops/random_sample/random_sample.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/include/ops/random_sample/random_sample.h -------------------------------------------------------------------------------- /include/ops/rearrange/rearrange.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/include/ops/rearrange/rearrange.h -------------------------------------------------------------------------------- /include/ops/relu/relu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/include/ops/relu/relu.h -------------------------------------------------------------------------------- /include/ops/rms_norm/rms_norm.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/include/ops/rms_norm/rms_norm.h -------------------------------------------------------------------------------- /include/ops/rotary_embedding/rotary_embedding.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/include/ops/rotary_embedding/rotary_embedding.h -------------------------------------------------------------------------------- /include/ops/swiglu/swiglu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/include/ops/swiglu/swiglu.h -------------------------------------------------------------------------------- /include/status.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/include/status.h -------------------------------------------------------------------------------- /include/tensor.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/include/tensor.h -------------------------------------------------------------------------------- /include/tensor/tensor_descriptor.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/include/tensor/tensor_descriptor.h -------------------------------------------------------------------------------- /operatorspy/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/operatorspy/__init__.py -------------------------------------------------------------------------------- /operatorspy/data_layout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/operatorspy/data_layout.py -------------------------------------------------------------------------------- /operatorspy/devices.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/operatorspy/devices.py -------------------------------------------------------------------------------- /operatorspy/liboperators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/operatorspy/liboperators.py -------------------------------------------------------------------------------- /operatorspy/tests/__init__.py: -------------------------------------------------------------------------------- 1 | import test_utils -------------------------------------------------------------------------------- /operatorspy/tests/add.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/operatorspy/tests/add.py -------------------------------------------------------------------------------- /operatorspy/tests/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/operatorspy/tests/attention.py -------------------------------------------------------------------------------- /operatorspy/tests/avg_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/operatorspy/tests/avg_pool.py -------------------------------------------------------------------------------- /operatorspy/tests/causal_softmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/operatorspy/tests/causal_softmax.py -------------------------------------------------------------------------------- /operatorspy/tests/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/operatorspy/tests/conv.py -------------------------------------------------------------------------------- /operatorspy/tests/expand.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/operatorspy/tests/expand.py -------------------------------------------------------------------------------- /operatorspy/tests/gemm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/operatorspy/tests/gemm.py -------------------------------------------------------------------------------- /operatorspy/tests/global_avg_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/operatorspy/tests/global_avg_pool.py -------------------------------------------------------------------------------- /operatorspy/tests/matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/operatorspy/tests/matmul.py -------------------------------------------------------------------------------- /operatorspy/tests/max_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/operatorspy/tests/max_pool.py -------------------------------------------------------------------------------- /operatorspy/tests/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/operatorspy/tests/mlp.py -------------------------------------------------------------------------------- /operatorspy/tests/random_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/operatorspy/tests/random_sample.py -------------------------------------------------------------------------------- /operatorspy/tests/rearrange.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/operatorspy/tests/rearrange.py -------------------------------------------------------------------------------- /operatorspy/tests/relu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/operatorspy/tests/relu.py -------------------------------------------------------------------------------- /operatorspy/tests/rms_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/operatorspy/tests/rms_norm.py -------------------------------------------------------------------------------- /operatorspy/tests/rotary_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/operatorspy/tests/rotary_embedding.py -------------------------------------------------------------------------------- /operatorspy/tests/swiglu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/operatorspy/tests/swiglu.py -------------------------------------------------------------------------------- /operatorspy/tests/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/operatorspy/tests/test_utils.py -------------------------------------------------------------------------------- /operatorspy/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/operatorspy/utils.py -------------------------------------------------------------------------------- /src/devices/ascend/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/devices/ascend/CMakeLists.txt -------------------------------------------------------------------------------- /src/devices/ascend/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/devices/ascend/Makefile -------------------------------------------------------------------------------- /src/devices/ascend/ascend_handle.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/devices/ascend/ascend_handle.cc -------------------------------------------------------------------------------- /src/devices/ascend/ascend_handle.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/devices/ascend/ascend_handle.h -------------------------------------------------------------------------------- /src/devices/ascend/common_ascend.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/devices/ascend/common_ascend.cc -------------------------------------------------------------------------------- /src/devices/ascend/common_ascend.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/devices/ascend/common_ascend.h -------------------------------------------------------------------------------- /src/devices/ascend/tensor_aclnn.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/devices/ascend/tensor_aclnn.cc -------------------------------------------------------------------------------- /src/devices/ascend/tensor_aclnn.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/devices/ascend/tensor_aclnn.h -------------------------------------------------------------------------------- /src/devices/bang/bang_handle.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/devices/bang/bang_handle.cc -------------------------------------------------------------------------------- /src/devices/bang/bang_handle.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/devices/bang/bang_handle.h -------------------------------------------------------------------------------- /src/devices/bang/common_bang.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/devices/bang/common_bang.h -------------------------------------------------------------------------------- /src/devices/cpu/common_cpu.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/devices/cpu/common_cpu.cc -------------------------------------------------------------------------------- /src/devices/cpu/common_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/devices/cpu/common_cpu.h -------------------------------------------------------------------------------- /src/devices/cpu/cpu_handle.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/devices/cpu/cpu_handle.cc -------------------------------------------------------------------------------- /src/devices/cpu/cpu_handle.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/devices/cpu/cpu_handle.h -------------------------------------------------------------------------------- /src/devices/cuda/common_cuda.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/devices/cuda/common_cuda.h -------------------------------------------------------------------------------- /src/devices/cuda/cuda_handle.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/devices/cuda/cuda_handle.cc -------------------------------------------------------------------------------- /src/devices/cuda/cuda_handle.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/devices/cuda/cuda_handle.h -------------------------------------------------------------------------------- /src/devices/handle.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/devices/handle.cc -------------------------------------------------------------------------------- /src/devices/maca/common_maca.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/devices/maca/common_maca.h -------------------------------------------------------------------------------- /src/devices/maca/maca_handle.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/devices/maca/maca_handle.cc -------------------------------------------------------------------------------- /src/devices/maca/maca_handle.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/devices/maca/maca_handle.h -------------------------------------------------------------------------------- /src/devices/musa/common_musa.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/devices/musa/common_musa.h -------------------------------------------------------------------------------- /src/devices/musa/musa_handle.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/devices/musa/musa_handle.cc -------------------------------------------------------------------------------- /src/devices/musa/musa_handle.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/devices/musa/musa_handle.h -------------------------------------------------------------------------------- /src/devices/musa/pool.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/devices/musa/pool.h -------------------------------------------------------------------------------- /src/devices/pool.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/devices/pool.h -------------------------------------------------------------------------------- /src/ops/add/cpu/add_cpu.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/add/cpu/add_cpu.cc -------------------------------------------------------------------------------- /src/ops/add/cpu/add_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/add/cpu/add_cpu.h -------------------------------------------------------------------------------- /src/ops/add/cuda/add.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/add/cuda/add.cc -------------------------------------------------------------------------------- /src/ops/add/cuda/add.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/add/cuda/add.cu -------------------------------------------------------------------------------- /src/ops/add/cuda/add.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/add/cuda/add.cuh -------------------------------------------------------------------------------- /src/ops/add/musa/add_musa.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/add/musa/add_musa.cc -------------------------------------------------------------------------------- /src/ops/add/musa/add_musa.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/add/musa/add_musa.h -------------------------------------------------------------------------------- /src/ops/add/musa/add_musa.mu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/add/musa/add_musa.mu -------------------------------------------------------------------------------- /src/ops/add/operator.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/add/operator.cc -------------------------------------------------------------------------------- /src/ops/attention/operator.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/attention/operator.cc -------------------------------------------------------------------------------- /src/ops/avg_pool/operator.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/avg_pool/operator.cc -------------------------------------------------------------------------------- /src/ops/causal_softmax/ascend/causal_softmax_aclnn.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/causal_softmax/ascend/causal_softmax_aclnn.cc -------------------------------------------------------------------------------- /src/ops/causal_softmax/ascend/causal_softmax_aclnn.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/causal_softmax/ascend/causal_softmax_aclnn.h -------------------------------------------------------------------------------- /src/ops/causal_softmax/bang/causal_softmax_bang.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/causal_softmax/bang/causal_softmax_bang.cc -------------------------------------------------------------------------------- /src/ops/causal_softmax/bang/causal_softmax_bang.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/causal_softmax/bang/causal_softmax_bang.h -------------------------------------------------------------------------------- /src/ops/causal_softmax/bang/causal_softmax_bang.mlu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/causal_softmax/bang/causal_softmax_bang.mlu -------------------------------------------------------------------------------- /src/ops/causal_softmax/bang/causal_softmax_cnnl.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/causal_softmax/bang/causal_softmax_cnnl.cc -------------------------------------------------------------------------------- /src/ops/causal_softmax/bang/causal_softmax_cnnl.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/causal_softmax/bang/causal_softmax_cnnl.h -------------------------------------------------------------------------------- /src/ops/causal_softmax/cpu/causal_softmax_cpu.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/causal_softmax/cpu/causal_softmax_cpu.cc -------------------------------------------------------------------------------- /src/ops/causal_softmax/cpu/causal_softmax_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/causal_softmax/cpu/causal_softmax_cpu.h -------------------------------------------------------------------------------- /src/ops/causal_softmax/cuda/causal_softmax.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/causal_softmax/cuda/causal_softmax.cc -------------------------------------------------------------------------------- /src/ops/causal_softmax/cuda/causal_softmax.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/causal_softmax/cuda/causal_softmax.cu -------------------------------------------------------------------------------- /src/ops/causal_softmax/cuda/causal_softmax.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/causal_softmax/cuda/causal_softmax.cuh -------------------------------------------------------------------------------- /src/ops/causal_softmax/maca/causal_softmax_maca.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/causal_softmax/maca/causal_softmax_maca.cc -------------------------------------------------------------------------------- /src/ops/causal_softmax/maca/causal_softmax_maca.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/causal_softmax/maca/causal_softmax_maca.h -------------------------------------------------------------------------------- /src/ops/causal_softmax/maca/causal_softmax_maca.maca: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/causal_softmax/maca/causal_softmax_maca.maca -------------------------------------------------------------------------------- /src/ops/causal_softmax/musa/causal_softmax_musa.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/causal_softmax/musa/causal_softmax_musa.cc -------------------------------------------------------------------------------- /src/ops/causal_softmax/musa/causal_softmax_musa.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/causal_softmax/musa/causal_softmax_musa.h -------------------------------------------------------------------------------- /src/ops/causal_softmax/musa/causal_softmax_musa.mu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/causal_softmax/musa/causal_softmax_musa.mu -------------------------------------------------------------------------------- /src/ops/causal_softmax/operator.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/causal_softmax/operator.cc -------------------------------------------------------------------------------- /src/ops/conv/cpu/conv_cpu.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/conv/cpu/conv_cpu.cc -------------------------------------------------------------------------------- /src/ops/conv/cpu/conv_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/conv/cpu/conv_cpu.h -------------------------------------------------------------------------------- /src/ops/conv/cuda/conv.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/conv/cuda/conv.cc -------------------------------------------------------------------------------- /src/ops/conv/cuda/conv.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/conv/cuda/conv.cu -------------------------------------------------------------------------------- /src/ops/conv/cuda/conv.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/conv/cuda/conv.cuh -------------------------------------------------------------------------------- /src/ops/conv/operator.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/conv/operator.cc -------------------------------------------------------------------------------- /src/ops/expand/cpu/expand_cpu.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/expand/cpu/expand_cpu.cc -------------------------------------------------------------------------------- /src/ops/expand/cpu/expand_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/expand/cpu/expand_cpu.h -------------------------------------------------------------------------------- /src/ops/expand/cuda/expand.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/expand/cuda/expand.cc -------------------------------------------------------------------------------- /src/ops/expand/cuda/expand.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/expand/cuda/expand.cu -------------------------------------------------------------------------------- /src/ops/expand/cuda/expand.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/expand/cuda/expand.cuh -------------------------------------------------------------------------------- /src/ops/expand/musa/expand_musa.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/expand/musa/expand_musa.cc -------------------------------------------------------------------------------- /src/ops/expand/musa/expand_musa.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/expand/musa/expand_musa.h -------------------------------------------------------------------------------- /src/ops/expand/musa/expand_musa.mu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/expand/musa/expand_musa.mu -------------------------------------------------------------------------------- /src/ops/expand/operator.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/expand/operator.cc -------------------------------------------------------------------------------- /src/ops/gemm/operator.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/gemm/operator.cc -------------------------------------------------------------------------------- /src/ops/global_avg_pool/cpu/global_avg_pool_cpu.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/global_avg_pool/cpu/global_avg_pool_cpu.cc -------------------------------------------------------------------------------- /src/ops/global_avg_pool/cpu/global_avg_pool_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/global_avg_pool/cpu/global_avg_pool_cpu.h -------------------------------------------------------------------------------- /src/ops/global_avg_pool/cuda/global_avg_pool.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/global_avg_pool/cuda/global_avg_pool.cc -------------------------------------------------------------------------------- /src/ops/global_avg_pool/cuda/global_avg_pool.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/global_avg_pool/cuda/global_avg_pool.cu -------------------------------------------------------------------------------- /src/ops/global_avg_pool/cuda/global_avg_pool.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/global_avg_pool/cuda/global_avg_pool.cuh -------------------------------------------------------------------------------- /src/ops/global_avg_pool/operator.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/global_avg_pool/operator.cc -------------------------------------------------------------------------------- /src/ops/matmul/ascend/matmul_aclnn.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/matmul/ascend/matmul_aclnn.cc -------------------------------------------------------------------------------- /src/ops/matmul/ascend/matmul_aclnn.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/matmul/ascend/matmul_aclnn.h -------------------------------------------------------------------------------- /src/ops/matmul/bang/matmul_cnnl.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/matmul/bang/matmul_cnnl.cc -------------------------------------------------------------------------------- /src/ops/matmul/bang/matmul_cnnl.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/matmul/bang/matmul_cnnl.h -------------------------------------------------------------------------------- /src/ops/matmul/blas.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/matmul/blas.h -------------------------------------------------------------------------------- /src/ops/matmul/cpu/matmul_cpu.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/matmul/cpu/matmul_cpu.cc -------------------------------------------------------------------------------- /src/ops/matmul/cpu/matmul_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/matmul/cpu/matmul_cpu.h -------------------------------------------------------------------------------- /src/ops/matmul/cuda/matmul_cuda.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/matmul/cuda/matmul_cuda.cc -------------------------------------------------------------------------------- /src/ops/matmul/cuda/matmul_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/matmul/cuda/matmul_cuda.cu -------------------------------------------------------------------------------- /src/ops/matmul/cuda/matmul_cuda.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/matmul/cuda/matmul_cuda.h -------------------------------------------------------------------------------- /src/ops/matmul/maca/matmul_maca.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/matmul/maca/matmul_maca.cc -------------------------------------------------------------------------------- /src/ops/matmul/maca/matmul_maca.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/matmul/maca/matmul_maca.h -------------------------------------------------------------------------------- /src/ops/matmul/maca/matmul_maca.maca: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/matmul/maca/matmul_maca.maca -------------------------------------------------------------------------------- /src/ops/matmul/musa/matmul_musa.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/matmul/musa/matmul_musa.cc -------------------------------------------------------------------------------- /src/ops/matmul/musa/matmul_musa.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/matmul/musa/matmul_musa.h -------------------------------------------------------------------------------- /src/ops/matmul/musa/matmul_musa.mu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/matmul/musa/matmul_musa.mu -------------------------------------------------------------------------------- /src/ops/matmul/operator.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/matmul/operator.cc -------------------------------------------------------------------------------- /src/ops/max_pool/operator.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/max_pool/operator.cc -------------------------------------------------------------------------------- /src/ops/mlp/operator.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/mlp/operator.cc -------------------------------------------------------------------------------- /src/ops/pooling/cpu/pooling_cpu.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/pooling/cpu/pooling_cpu.cc -------------------------------------------------------------------------------- /src/ops/pooling/cpu/pooling_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/pooling/cpu/pooling_cpu.h -------------------------------------------------------------------------------- /src/ops/pooling/cuda/pooling.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/pooling/cuda/pooling.cc -------------------------------------------------------------------------------- /src/ops/pooling/cuda/pooling.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/pooling/cuda/pooling.cu -------------------------------------------------------------------------------- /src/ops/pooling/cuda/pooling.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/pooling/cuda/pooling.cuh -------------------------------------------------------------------------------- /src/ops/pooling/operator.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/pooling/operator.cc -------------------------------------------------------------------------------- /src/ops/pooling/pooling.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/pooling/pooling.h -------------------------------------------------------------------------------- /src/ops/random_sample/ascend/random_sample.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/random_sample/ascend/random_sample.cc -------------------------------------------------------------------------------- /src/ops/random_sample/ascend/random_sample.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/random_sample/ascend/random_sample.h -------------------------------------------------------------------------------- /src/ops/random_sample/ascend/random_sample_kernel.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/random_sample/ascend/random_sample_kernel.cpp -------------------------------------------------------------------------------- /src/ops/random_sample/bang/random_sample_bang.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/random_sample/bang/random_sample_bang.cc -------------------------------------------------------------------------------- /src/ops/random_sample/bang/random_sample_bang.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/random_sample/bang/random_sample_bang.h -------------------------------------------------------------------------------- /src/ops/random_sample/bang/random_sample_bang.mlu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/random_sample/bang/random_sample_bang.mlu -------------------------------------------------------------------------------- /src/ops/random_sample/cpu/random_sample.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/random_sample/cpu/random_sample.cc -------------------------------------------------------------------------------- /src/ops/random_sample/cpu/random_sample_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/random_sample/cpu/random_sample_cpu.h -------------------------------------------------------------------------------- /src/ops/random_sample/cuda/random_sample.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/random_sample/cuda/random_sample.cu -------------------------------------------------------------------------------- /src/ops/random_sample/cuda/random_sample.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/random_sample/cuda/random_sample.cuh -------------------------------------------------------------------------------- /src/ops/random_sample/cuda/random_sample_cuda.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/random_sample/cuda/random_sample_cuda.cc -------------------------------------------------------------------------------- /src/ops/random_sample/maca/random_sample_maca.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/random_sample/maca/random_sample_maca.cc -------------------------------------------------------------------------------- /src/ops/random_sample/maca/random_sample_maca.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/random_sample/maca/random_sample_maca.h -------------------------------------------------------------------------------- /src/ops/random_sample/maca/random_sample_maca.maca: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/random_sample/maca/random_sample_maca.maca -------------------------------------------------------------------------------- /src/ops/random_sample/musa/random_sample_musa.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/random_sample/musa/random_sample_musa.cc -------------------------------------------------------------------------------- /src/ops/random_sample/musa/random_sample_musa.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/random_sample/musa/random_sample_musa.h -------------------------------------------------------------------------------- /src/ops/random_sample/musa/random_sample_musa.mu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/random_sample/musa/random_sample_musa.mu -------------------------------------------------------------------------------- /src/ops/random_sample/operator.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/random_sample/operator.cc -------------------------------------------------------------------------------- /src/ops/rearrange/ascend/rearrange_aclnn.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rearrange/ascend/rearrange_aclnn.cc -------------------------------------------------------------------------------- /src/ops/rearrange/ascend/rearrange_aclnn.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rearrange/ascend/rearrange_aclnn.h -------------------------------------------------------------------------------- /src/ops/rearrange/bang/rearrange_bang.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rearrange/bang/rearrange_bang.cc -------------------------------------------------------------------------------- /src/ops/rearrange/bang/rearrange_bang.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rearrange/bang/rearrange_bang.h -------------------------------------------------------------------------------- /src/ops/rearrange/bang/rearrange_bang.mlu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rearrange/bang/rearrange_bang.mlu -------------------------------------------------------------------------------- /src/ops/rearrange/cpu/rearrange_cpu.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rearrange/cpu/rearrange_cpu.cc -------------------------------------------------------------------------------- /src/ops/rearrange/cpu/rearrange_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rearrange/cpu/rearrange_cpu.h -------------------------------------------------------------------------------- /src/ops/rearrange/cuda/rearrange.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rearrange/cuda/rearrange.cc -------------------------------------------------------------------------------- /src/ops/rearrange/cuda/rearrange.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rearrange/cuda/rearrange.cu -------------------------------------------------------------------------------- /src/ops/rearrange/cuda/rearrange.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rearrange/cuda/rearrange.cuh -------------------------------------------------------------------------------- /src/ops/rearrange/maca/rearrange_maca.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rearrange/maca/rearrange_maca.cc -------------------------------------------------------------------------------- /src/ops/rearrange/maca/rearrange_maca.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rearrange/maca/rearrange_maca.h -------------------------------------------------------------------------------- /src/ops/rearrange/maca/rearrange_maca.maca: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rearrange/maca/rearrange_maca.maca -------------------------------------------------------------------------------- /src/ops/rearrange/musa/rearrange_musa.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rearrange/musa/rearrange_musa.cc -------------------------------------------------------------------------------- /src/ops/rearrange/musa/rearrange_musa.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rearrange/musa/rearrange_musa.h -------------------------------------------------------------------------------- /src/ops/rearrange/musa/rearrange_musa.mu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rearrange/musa/rearrange_musa.mu -------------------------------------------------------------------------------- /src/ops/rearrange/operator.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rearrange/operator.cc -------------------------------------------------------------------------------- /src/ops/relu/cpu/relu_cpu.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/relu/cpu/relu_cpu.cc -------------------------------------------------------------------------------- /src/ops/relu/cpu/relu_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/relu/cpu/relu_cpu.h -------------------------------------------------------------------------------- /src/ops/relu/cuda/relu.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/relu/cuda/relu.cc -------------------------------------------------------------------------------- /src/ops/relu/cuda/relu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/relu/cuda/relu.cu -------------------------------------------------------------------------------- /src/ops/relu/cuda/relu.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/relu/cuda/relu.cuh -------------------------------------------------------------------------------- /src/ops/relu/musa/relu_musa.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/relu/musa/relu_musa.cc -------------------------------------------------------------------------------- /src/ops/relu/musa/relu_musa.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/relu/musa/relu_musa.h -------------------------------------------------------------------------------- /src/ops/relu/musa/relu_musa.mu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/relu/musa/relu_musa.mu -------------------------------------------------------------------------------- /src/ops/relu/operator.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/relu/operator.cc -------------------------------------------------------------------------------- /src/ops/rms_norm/ascend/rms_norm_aclnn.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rms_norm/ascend/rms_norm_aclnn.cc -------------------------------------------------------------------------------- /src/ops/rms_norm/ascend/rms_norm_aclnn.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rms_norm/ascend/rms_norm_aclnn.h -------------------------------------------------------------------------------- /src/ops/rms_norm/bang/rms_norm_bang.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rms_norm/bang/rms_norm_bang.cc -------------------------------------------------------------------------------- /src/ops/rms_norm/bang/rms_norm_bang.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rms_norm/bang/rms_norm_bang.h -------------------------------------------------------------------------------- /src/ops/rms_norm/bang/rms_norm_bang.mlu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rms_norm/bang/rms_norm_bang.mlu -------------------------------------------------------------------------------- /src/ops/rms_norm/cpu/rms_norm_cpu.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rms_norm/cpu/rms_norm_cpu.cc -------------------------------------------------------------------------------- /src/ops/rms_norm/cpu/rms_norm_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rms_norm/cpu/rms_norm_cpu.h -------------------------------------------------------------------------------- /src/ops/rms_norm/cuda/rms_norm.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rms_norm/cuda/rms_norm.cc -------------------------------------------------------------------------------- /src/ops/rms_norm/cuda/rms_norm.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rms_norm/cuda/rms_norm.cu -------------------------------------------------------------------------------- /src/ops/rms_norm/cuda/rms_norm.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rms_norm/cuda/rms_norm.cuh -------------------------------------------------------------------------------- /src/ops/rms_norm/maca/rms_norm_maca.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rms_norm/maca/rms_norm_maca.cc -------------------------------------------------------------------------------- /src/ops/rms_norm/maca/rms_norm_maca.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rms_norm/maca/rms_norm_maca.h -------------------------------------------------------------------------------- /src/ops/rms_norm/maca/rms_norm_maca.maca: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rms_norm/maca/rms_norm_maca.maca -------------------------------------------------------------------------------- /src/ops/rms_norm/musa/rms_norm_musa.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rms_norm/musa/rms_norm_musa.cc -------------------------------------------------------------------------------- /src/ops/rms_norm/musa/rms_norm_musa.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rms_norm/musa/rms_norm_musa.h -------------------------------------------------------------------------------- /src/ops/rms_norm/musa/rms_norm_musa.mu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rms_norm/musa/rms_norm_musa.mu -------------------------------------------------------------------------------- /src/ops/rms_norm/operator.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rms_norm/operator.cc -------------------------------------------------------------------------------- /src/ops/rotary_embedding/ascend/rotary_embedding.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rotary_embedding/ascend/rotary_embedding.cc -------------------------------------------------------------------------------- /src/ops/rotary_embedding/ascend/rotary_embedding.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rotary_embedding/ascend/rotary_embedding.h -------------------------------------------------------------------------------- /src/ops/rotary_embedding/ascend/rotary_embedding_kernel.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rotary_embedding/ascend/rotary_embedding_kernel.cpp -------------------------------------------------------------------------------- /src/ops/rotary_embedding/bang/rotary_embedding_bang.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rotary_embedding/bang/rotary_embedding_bang.cc -------------------------------------------------------------------------------- /src/ops/rotary_embedding/bang/rotary_embedding_bang.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rotary_embedding/bang/rotary_embedding_bang.h -------------------------------------------------------------------------------- /src/ops/rotary_embedding/bang/rotary_embedding_bang.mlu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rotary_embedding/bang/rotary_embedding_bang.mlu -------------------------------------------------------------------------------- /src/ops/rotary_embedding/cpu/rotary_embedding_cpu.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rotary_embedding/cpu/rotary_embedding_cpu.cc -------------------------------------------------------------------------------- /src/ops/rotary_embedding/cpu/rotary_embedding_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rotary_embedding/cpu/rotary_embedding_cpu.h -------------------------------------------------------------------------------- /src/ops/rotary_embedding/cuda/rotary_embedding.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rotary_embedding/cuda/rotary_embedding.cc -------------------------------------------------------------------------------- /src/ops/rotary_embedding/cuda/rotary_embedding.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rotary_embedding/cuda/rotary_embedding.cu -------------------------------------------------------------------------------- /src/ops/rotary_embedding/cuda/rotary_embedding.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rotary_embedding/cuda/rotary_embedding.cuh -------------------------------------------------------------------------------- /src/ops/rotary_embedding/maca/rotary_embedding_maca.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rotary_embedding/maca/rotary_embedding_maca.cc -------------------------------------------------------------------------------- /src/ops/rotary_embedding/maca/rotary_embedding_maca.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rotary_embedding/maca/rotary_embedding_maca.h -------------------------------------------------------------------------------- /src/ops/rotary_embedding/maca/rotary_embedding_maca.maca: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rotary_embedding/maca/rotary_embedding_maca.maca -------------------------------------------------------------------------------- /src/ops/rotary_embedding/musa/rotary_embedding_musa.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rotary_embedding/musa/rotary_embedding_musa.cc -------------------------------------------------------------------------------- /src/ops/rotary_embedding/musa/rotary_embedding_musa.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rotary_embedding/musa/rotary_embedding_musa.h -------------------------------------------------------------------------------- /src/ops/rotary_embedding/musa/rotary_embedding_musa.mu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rotary_embedding/musa/rotary_embedding_musa.mu -------------------------------------------------------------------------------- /src/ops/rotary_embedding/operator.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/rotary_embedding/operator.cc -------------------------------------------------------------------------------- /src/ops/swiglu/ascend/swiglu.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/swiglu/ascend/swiglu.cc -------------------------------------------------------------------------------- /src/ops/swiglu/ascend/swiglu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/swiglu/ascend/swiglu.h -------------------------------------------------------------------------------- /src/ops/swiglu/ascend/swiglu_kernel.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/swiglu/ascend/swiglu_kernel.cpp -------------------------------------------------------------------------------- /src/ops/swiglu/bang/swiglu_bang.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/swiglu/bang/swiglu_bang.cc -------------------------------------------------------------------------------- /src/ops/swiglu/bang/swiglu_bang.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/swiglu/bang/swiglu_bang.h -------------------------------------------------------------------------------- /src/ops/swiglu/bang/swiglu_bang.mlu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/swiglu/bang/swiglu_bang.mlu -------------------------------------------------------------------------------- /src/ops/swiglu/cpu/swiglu_cpu.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/swiglu/cpu/swiglu_cpu.cc -------------------------------------------------------------------------------- /src/ops/swiglu/cpu/swiglu_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/swiglu/cpu/swiglu_cpu.h -------------------------------------------------------------------------------- /src/ops/swiglu/cuda/swiglu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/swiglu/cuda/swiglu.cu -------------------------------------------------------------------------------- /src/ops/swiglu/cuda/swiglu.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/swiglu/cuda/swiglu.cuh -------------------------------------------------------------------------------- /src/ops/swiglu/cuda/swiglu_cuda.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/swiglu/cuda/swiglu_cuda.cc -------------------------------------------------------------------------------- /src/ops/swiglu/maca/swiglu_maca.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/swiglu/maca/swiglu_maca.cc -------------------------------------------------------------------------------- /src/ops/swiglu/maca/swiglu_maca.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/swiglu/maca/swiglu_maca.h -------------------------------------------------------------------------------- /src/ops/swiglu/maca/swiglu_maca.maca: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/swiglu/maca/swiglu_maca.maca -------------------------------------------------------------------------------- /src/ops/swiglu/musa/swiglu.mu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/swiglu/musa/swiglu.mu -------------------------------------------------------------------------------- /src/ops/swiglu/musa/swiglu_musa.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/swiglu/musa/swiglu_musa.cc -------------------------------------------------------------------------------- /src/ops/swiglu/musa/swiglu_musa.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/swiglu/musa/swiglu_musa.h -------------------------------------------------------------------------------- /src/ops/swiglu/operator.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/swiglu/operator.cc -------------------------------------------------------------------------------- /src/ops/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/ops/utils.h -------------------------------------------------------------------------------- /src/tensor/tensor_descriptor.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/src/tensor/tensor_descriptor.cc -------------------------------------------------------------------------------- /xmake.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InfiniTensor/operators/HEAD/xmake.lua --------------------------------------------------------------------------------