├── .clang-format ├── .gitignore ├── .style.yapf ├── 3rd └── dbg │ └── dbg.h ├── CMakeLists.txt ├── LICENSE ├── README.md ├── bench ├── attn.mlir ├── ea.mlir ├── h2o.mlir ├── keyformer.mlir ├── roco.mlir └── sima.mlir ├── bin ├── CMakeLists.txt ├── asuka-kernel.cpp ├── asuka-opt.cpp └── asuka-translate.cpp ├── cases ├── attn.py ├── attn_h2o.py ├── kernels │ ├── attn.py │ ├── corm.py │ ├── gemma2.py │ ├── h2o.py │ ├── kf.py │ ├── roco.py │ └── snapkv.py └── run.py ├── cmake └── QueryPythonForPybind11.cmake ├── ffi ├── ffi.cpp ├── ffi.h ├── ir.cpp └── passes.cpp ├── include ├── CMakeLists.txt └── asuka │ ├── Analysis │ └── Parallelism.h │ ├── CMakeLists.txt │ ├── Conversion │ ├── AsukaToAsukaTriton │ │ ├── CMakeLists.txt │ │ ├── Passes.h │ │ └── Passes.td │ └── CMakeLists.txt │ ├── Dialect │ ├── Asuka │ │ ├── CMakeLists.txt │ │ ├── IR │ │ │ ├── AsukaDialect.h │ │ │ ├── AsukaDialect.td │ │ │ ├── AsukaEnums.td │ │ │ ├── AsukaInterfaces.td │ │ │ ├── AsukaOps.td │ │ │ ├── AsukaTypes.h │ │ │ ├── AsukaTypes.td │ │ │ └── CMakeLists.txt │ │ └── Transforms │ │ │ ├── CMakeLists.txt │ │ │ ├── Passes.h │ │ │ └── Passes.td │ ├── AsukaTriton │ │ ├── CMakeLists.txt │ │ ├── IR │ │ │ ├── AsukaTritonDialect.h │ │ │ ├── AsukaTritonDialect.td │ │ │ ├── AsukaTritonOps.td │ │ │ ├── AsukaTritonTypes.h │ │ │ ├── AsukaTritonTypes.td │ │ │ └── CMakeLists.txt │ │ └── Transforms │ │ │ ├── CMakeLists.txt │ │ │ ├── Passes.h │ │ │ └── Passes.td │ └── CMakeLists.txt │ └── Translate │ └── translate.h ├── lib ├── Analysis │ ├── CMakeLists.txt │ └── Parallelism.cpp ├── CMakeLists.txt ├── Conversion │ ├── AsukaToAsukaTriton │ │ ├── AsukaToAsukaTritonPass.cpp │ │ └── CMakeLists.txt │ └── CMakeLists.txt ├── Dialect │ ├── Asuka │ │ ├── CMakeLists.txt │ │ ├── IR │ │ │ ├── AsukaOps.cpp │ │ │ ├── AsukaTypes.cpp │ │ │ └── CMakeLists.txt │ │ └── Transforms │ │ │ ├── AnnotateParallelism.cpp │ │ │ ├── BroadcastTransform.cpp │ │ │ ├── CMakeLists.txt │ │ │ ├── DynamicFor.cpp │ │ │ ├── EquivalentTransform.cpp │ │ │ ├── EraseTypeInKernel.cpp │ │ │ ├── LowerComplexReduce.cpp │ │ │ ├── MulScalarHoisting.cpp │ │ │ ├── Parallelize.cpp │ │ │ ├── PermuteHoisting.cpp │ │ │ ├── RecoverTypeInKernel.cpp │ │ │ ├── ReplaceExpAndLog.cpp │ │ │ ├── Simplify.cpp │ │ │ ├── Tiling.cpp │ │ │ ├── ToMask.cpp │ │ │ └── UserReplicate.cpp │ ├── AsukaTriton │ │ ├── CMakeLists.txt │ │ ├── IR │ │ │ ├── AsukaTritonOps.cpp │ │ │ ├── AsukaTritonTypes.cpp │ │ │ └── CMakeLists.txt │ │ └── Transforms │ │ │ ├── CMakeLists.txt │ │ │ ├── SqueezeBlock.cpp │ │ │ └── UserReplicate.cpp │ └── CMakeLists.txt └── Translate │ ├── CMakeLists.txt │ └── to_triton.cpp ├── pyproject.toml ├── python └── asuka │ ├── __init__.py │ ├── backend │ ├── __init__.py │ ├── builder.py │ └── tensorrt │ │ ├── __init__.py │ │ ├── builder.py │ │ └── utils.py │ ├── partition │ ├── config.py │ ├── connected.py │ ├── convex.py │ └── kernel.py │ ├── subgraph.py │ ├── transform │ ├── __init__.py │ ├── common.py │ ├── fission.py │ └── utils.py │ ├── translate.py │ └── utils.py ├── qiyuan_env.sh ├── rebuild.sh ├── script ├── format.sh ├── llvm_build.sh └── qiyuan_dep_build.sh └── yes_env.sh /.clang-format: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/.clang-format -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/.gitignore -------------------------------------------------------------------------------- /.style.yapf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/.style.yapf -------------------------------------------------------------------------------- /3rd/dbg/dbg.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/3rd/dbg/dbg.h -------------------------------------------------------------------------------- /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/README.md -------------------------------------------------------------------------------- /bench/attn.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/bench/attn.mlir -------------------------------------------------------------------------------- /bench/ea.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/bench/ea.mlir -------------------------------------------------------------------------------- /bench/h2o.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/bench/h2o.mlir -------------------------------------------------------------------------------- /bench/keyformer.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/bench/keyformer.mlir -------------------------------------------------------------------------------- /bench/roco.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/bench/roco.mlir -------------------------------------------------------------------------------- /bench/sima.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/bench/sima.mlir -------------------------------------------------------------------------------- /bin/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/bin/CMakeLists.txt -------------------------------------------------------------------------------- /bin/asuka-kernel.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/bin/asuka-kernel.cpp -------------------------------------------------------------------------------- /bin/asuka-opt.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/bin/asuka-opt.cpp -------------------------------------------------------------------------------- /bin/asuka-translate.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/bin/asuka-translate.cpp -------------------------------------------------------------------------------- /cases/attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/cases/attn.py -------------------------------------------------------------------------------- /cases/attn_h2o.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/cases/attn_h2o.py -------------------------------------------------------------------------------- /cases/kernels/attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/cases/kernels/attn.py -------------------------------------------------------------------------------- /cases/kernels/corm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/cases/kernels/corm.py -------------------------------------------------------------------------------- /cases/kernels/gemma2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/cases/kernels/gemma2.py -------------------------------------------------------------------------------- /cases/kernels/h2o.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/cases/kernels/h2o.py -------------------------------------------------------------------------------- /cases/kernels/kf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/cases/kernels/kf.py -------------------------------------------------------------------------------- /cases/kernels/roco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/cases/kernels/roco.py -------------------------------------------------------------------------------- /cases/kernels/snapkv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/cases/kernels/snapkv.py -------------------------------------------------------------------------------- /cases/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/cases/run.py -------------------------------------------------------------------------------- /cmake/QueryPythonForPybind11.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/cmake/QueryPythonForPybind11.cmake -------------------------------------------------------------------------------- /ffi/ffi.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/ffi/ffi.cpp -------------------------------------------------------------------------------- /ffi/ffi.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/ffi/ffi.h -------------------------------------------------------------------------------- /ffi/ir.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/ffi/ir.cpp -------------------------------------------------------------------------------- /ffi/passes.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/ffi/passes.cpp -------------------------------------------------------------------------------- /include/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(asuka) -------------------------------------------------------------------------------- /include/asuka/Analysis/Parallelism.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/include/asuka/Analysis/Parallelism.h -------------------------------------------------------------------------------- /include/asuka/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/include/asuka/CMakeLists.txt -------------------------------------------------------------------------------- /include/asuka/Conversion/AsukaToAsukaTriton/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/include/asuka/Conversion/AsukaToAsukaTriton/CMakeLists.txt -------------------------------------------------------------------------------- /include/asuka/Conversion/AsukaToAsukaTriton/Passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/include/asuka/Conversion/AsukaToAsukaTriton/Passes.h -------------------------------------------------------------------------------- /include/asuka/Conversion/AsukaToAsukaTriton/Passes.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/include/asuka/Conversion/AsukaToAsukaTriton/Passes.td -------------------------------------------------------------------------------- /include/asuka/Conversion/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/include/asuka/Conversion/CMakeLists.txt -------------------------------------------------------------------------------- /include/asuka/Dialect/Asuka/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/include/asuka/Dialect/Asuka/CMakeLists.txt -------------------------------------------------------------------------------- /include/asuka/Dialect/Asuka/IR/AsukaDialect.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/include/asuka/Dialect/Asuka/IR/AsukaDialect.h -------------------------------------------------------------------------------- /include/asuka/Dialect/Asuka/IR/AsukaDialect.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/include/asuka/Dialect/Asuka/IR/AsukaDialect.td -------------------------------------------------------------------------------- /include/asuka/Dialect/Asuka/IR/AsukaEnums.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/include/asuka/Dialect/Asuka/IR/AsukaEnums.td -------------------------------------------------------------------------------- /include/asuka/Dialect/Asuka/IR/AsukaInterfaces.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/include/asuka/Dialect/Asuka/IR/AsukaInterfaces.td -------------------------------------------------------------------------------- /include/asuka/Dialect/Asuka/IR/AsukaOps.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/include/asuka/Dialect/Asuka/IR/AsukaOps.td -------------------------------------------------------------------------------- /include/asuka/Dialect/Asuka/IR/AsukaTypes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/include/asuka/Dialect/Asuka/IR/AsukaTypes.h -------------------------------------------------------------------------------- /include/asuka/Dialect/Asuka/IR/AsukaTypes.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/include/asuka/Dialect/Asuka/IR/AsukaTypes.td -------------------------------------------------------------------------------- /include/asuka/Dialect/Asuka/IR/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/include/asuka/Dialect/Asuka/IR/CMakeLists.txt -------------------------------------------------------------------------------- /include/asuka/Dialect/Asuka/Transforms/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/include/asuka/Dialect/Asuka/Transforms/CMakeLists.txt -------------------------------------------------------------------------------- /include/asuka/Dialect/Asuka/Transforms/Passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/include/asuka/Dialect/Asuka/Transforms/Passes.h -------------------------------------------------------------------------------- /include/asuka/Dialect/Asuka/Transforms/Passes.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/include/asuka/Dialect/Asuka/Transforms/Passes.td -------------------------------------------------------------------------------- /include/asuka/Dialect/AsukaTriton/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/include/asuka/Dialect/AsukaTriton/CMakeLists.txt -------------------------------------------------------------------------------- /include/asuka/Dialect/AsukaTriton/IR/AsukaTritonDialect.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/include/asuka/Dialect/AsukaTriton/IR/AsukaTritonDialect.h -------------------------------------------------------------------------------- /include/asuka/Dialect/AsukaTriton/IR/AsukaTritonDialect.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/include/asuka/Dialect/AsukaTriton/IR/AsukaTritonDialect.td -------------------------------------------------------------------------------- /include/asuka/Dialect/AsukaTriton/IR/AsukaTritonOps.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/include/asuka/Dialect/AsukaTriton/IR/AsukaTritonOps.td -------------------------------------------------------------------------------- /include/asuka/Dialect/AsukaTriton/IR/AsukaTritonTypes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/include/asuka/Dialect/AsukaTriton/IR/AsukaTritonTypes.h -------------------------------------------------------------------------------- /include/asuka/Dialect/AsukaTriton/IR/AsukaTritonTypes.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/include/asuka/Dialect/AsukaTriton/IR/AsukaTritonTypes.td -------------------------------------------------------------------------------- /include/asuka/Dialect/AsukaTriton/IR/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/include/asuka/Dialect/AsukaTriton/IR/CMakeLists.txt -------------------------------------------------------------------------------- /include/asuka/Dialect/AsukaTriton/Transforms/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/include/asuka/Dialect/AsukaTriton/Transforms/CMakeLists.txt -------------------------------------------------------------------------------- /include/asuka/Dialect/AsukaTriton/Transforms/Passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/include/asuka/Dialect/AsukaTriton/Transforms/Passes.h -------------------------------------------------------------------------------- /include/asuka/Dialect/AsukaTriton/Transforms/Passes.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/include/asuka/Dialect/AsukaTriton/Transforms/Passes.td -------------------------------------------------------------------------------- /include/asuka/Dialect/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/include/asuka/Dialect/CMakeLists.txt -------------------------------------------------------------------------------- /include/asuka/Translate/translate.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/include/asuka/Translate/translate.h -------------------------------------------------------------------------------- /lib/Analysis/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/lib/Analysis/CMakeLists.txt -------------------------------------------------------------------------------- /lib/Analysis/Parallelism.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/lib/Analysis/Parallelism.cpp -------------------------------------------------------------------------------- /lib/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/lib/CMakeLists.txt -------------------------------------------------------------------------------- /lib/Conversion/AsukaToAsukaTriton/AsukaToAsukaTritonPass.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/lib/Conversion/AsukaToAsukaTriton/AsukaToAsukaTritonPass.cpp -------------------------------------------------------------------------------- /lib/Conversion/AsukaToAsukaTriton/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/lib/Conversion/AsukaToAsukaTriton/CMakeLists.txt -------------------------------------------------------------------------------- /lib/Conversion/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/lib/Conversion/CMakeLists.txt -------------------------------------------------------------------------------- /lib/Dialect/Asuka/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/lib/Dialect/Asuka/CMakeLists.txt -------------------------------------------------------------------------------- /lib/Dialect/Asuka/IR/AsukaOps.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/lib/Dialect/Asuka/IR/AsukaOps.cpp -------------------------------------------------------------------------------- /lib/Dialect/Asuka/IR/AsukaTypes.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/lib/Dialect/Asuka/IR/AsukaTypes.cpp -------------------------------------------------------------------------------- /lib/Dialect/Asuka/IR/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/lib/Dialect/Asuka/IR/CMakeLists.txt -------------------------------------------------------------------------------- /lib/Dialect/Asuka/Transforms/AnnotateParallelism.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/lib/Dialect/Asuka/Transforms/AnnotateParallelism.cpp -------------------------------------------------------------------------------- /lib/Dialect/Asuka/Transforms/BroadcastTransform.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/lib/Dialect/Asuka/Transforms/BroadcastTransform.cpp -------------------------------------------------------------------------------- /lib/Dialect/Asuka/Transforms/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/lib/Dialect/Asuka/Transforms/CMakeLists.txt -------------------------------------------------------------------------------- /lib/Dialect/Asuka/Transforms/DynamicFor.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/lib/Dialect/Asuka/Transforms/DynamicFor.cpp -------------------------------------------------------------------------------- /lib/Dialect/Asuka/Transforms/EquivalentTransform.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/lib/Dialect/Asuka/Transforms/EquivalentTransform.cpp -------------------------------------------------------------------------------- /lib/Dialect/Asuka/Transforms/EraseTypeInKernel.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/lib/Dialect/Asuka/Transforms/EraseTypeInKernel.cpp -------------------------------------------------------------------------------- /lib/Dialect/Asuka/Transforms/LowerComplexReduce.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/lib/Dialect/Asuka/Transforms/LowerComplexReduce.cpp -------------------------------------------------------------------------------- /lib/Dialect/Asuka/Transforms/MulScalarHoisting.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/lib/Dialect/Asuka/Transforms/MulScalarHoisting.cpp -------------------------------------------------------------------------------- /lib/Dialect/Asuka/Transforms/Parallelize.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/lib/Dialect/Asuka/Transforms/Parallelize.cpp -------------------------------------------------------------------------------- /lib/Dialect/Asuka/Transforms/PermuteHoisting.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/lib/Dialect/Asuka/Transforms/PermuteHoisting.cpp -------------------------------------------------------------------------------- /lib/Dialect/Asuka/Transforms/RecoverTypeInKernel.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/lib/Dialect/Asuka/Transforms/RecoverTypeInKernel.cpp -------------------------------------------------------------------------------- /lib/Dialect/Asuka/Transforms/ReplaceExpAndLog.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/lib/Dialect/Asuka/Transforms/ReplaceExpAndLog.cpp -------------------------------------------------------------------------------- /lib/Dialect/Asuka/Transforms/Simplify.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/lib/Dialect/Asuka/Transforms/Simplify.cpp -------------------------------------------------------------------------------- /lib/Dialect/Asuka/Transforms/Tiling.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/lib/Dialect/Asuka/Transforms/Tiling.cpp -------------------------------------------------------------------------------- /lib/Dialect/Asuka/Transforms/ToMask.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/lib/Dialect/Asuka/Transforms/ToMask.cpp -------------------------------------------------------------------------------- /lib/Dialect/Asuka/Transforms/UserReplicate.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/lib/Dialect/Asuka/Transforms/UserReplicate.cpp -------------------------------------------------------------------------------- /lib/Dialect/AsukaTriton/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/lib/Dialect/AsukaTriton/CMakeLists.txt -------------------------------------------------------------------------------- /lib/Dialect/AsukaTriton/IR/AsukaTritonOps.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/lib/Dialect/AsukaTriton/IR/AsukaTritonOps.cpp -------------------------------------------------------------------------------- /lib/Dialect/AsukaTriton/IR/AsukaTritonTypes.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/lib/Dialect/AsukaTriton/IR/AsukaTritonTypes.cpp -------------------------------------------------------------------------------- /lib/Dialect/AsukaTriton/IR/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/lib/Dialect/AsukaTriton/IR/CMakeLists.txt -------------------------------------------------------------------------------- /lib/Dialect/AsukaTriton/Transforms/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/lib/Dialect/AsukaTriton/Transforms/CMakeLists.txt -------------------------------------------------------------------------------- /lib/Dialect/AsukaTriton/Transforms/SqueezeBlock.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/lib/Dialect/AsukaTriton/Transforms/SqueezeBlock.cpp -------------------------------------------------------------------------------- /lib/Dialect/AsukaTriton/Transforms/UserReplicate.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/lib/Dialect/AsukaTriton/Transforms/UserReplicate.cpp -------------------------------------------------------------------------------- /lib/Dialect/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/lib/Dialect/CMakeLists.txt -------------------------------------------------------------------------------- /lib/Translate/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/lib/Translate/CMakeLists.txt -------------------------------------------------------------------------------- /lib/Translate/to_triton.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/lib/Translate/to_triton.cpp -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/pyproject.toml -------------------------------------------------------------------------------- /python/asuka/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/python/asuka/__init__.py -------------------------------------------------------------------------------- /python/asuka/backend/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/python/asuka/backend/__init__.py -------------------------------------------------------------------------------- /python/asuka/backend/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/python/asuka/backend/builder.py -------------------------------------------------------------------------------- /python/asuka/backend/tensorrt/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/python/asuka/backend/tensorrt/__init__.py -------------------------------------------------------------------------------- /python/asuka/backend/tensorrt/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/python/asuka/backend/tensorrt/builder.py -------------------------------------------------------------------------------- /python/asuka/backend/tensorrt/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/python/asuka/backend/tensorrt/utils.py -------------------------------------------------------------------------------- /python/asuka/partition/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/python/asuka/partition/config.py -------------------------------------------------------------------------------- /python/asuka/partition/connected.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/python/asuka/partition/connected.py -------------------------------------------------------------------------------- /python/asuka/partition/convex.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /python/asuka/partition/kernel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/python/asuka/partition/kernel.py -------------------------------------------------------------------------------- /python/asuka/subgraph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/python/asuka/subgraph.py -------------------------------------------------------------------------------- /python/asuka/transform/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/python/asuka/transform/__init__.py -------------------------------------------------------------------------------- /python/asuka/transform/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/python/asuka/transform/common.py -------------------------------------------------------------------------------- /python/asuka/transform/fission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/python/asuka/transform/fission.py -------------------------------------------------------------------------------- /python/asuka/transform/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/python/asuka/transform/utils.py -------------------------------------------------------------------------------- /python/asuka/translate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/python/asuka/translate.py -------------------------------------------------------------------------------- /python/asuka/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/python/asuka/utils.py -------------------------------------------------------------------------------- /qiyuan_env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/qiyuan_env.sh -------------------------------------------------------------------------------- /rebuild.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/rebuild.sh -------------------------------------------------------------------------------- /script/format.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/script/format.sh -------------------------------------------------------------------------------- /script/llvm_build.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/script/llvm_build.sh -------------------------------------------------------------------------------- /script/qiyuan_dep_build.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/script/qiyuan_dep_build.sh -------------------------------------------------------------------------------- /yes_env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monellz/FlashTensor/HEAD/yes_env.sh --------------------------------------------------------------------------------