├── .clang-format ├── .github ├── ISSUE_TEMPLATE │ ├── bug.md │ ├── feature-request.md │ └── question.md └── workflows │ ├── compiler-ci.yaml │ ├── daily_ci.yaml │ ├── e2e_test.yaml │ ├── format-check.yaml │ ├── onnx-frontend-ci.yaml │ ├── runtime-ci.yaml │ ├── tf-frontend-ci.yaml │ └── torch-frontend-ci.yaml ├── .gitignore ├── .gitmodules ├── CITATION.cff ├── CONTRIBUTING.md ├── LICENSE ├── NOTICE ├── README-zh_cn.md ├── README.md ├── compiler ├── .gitignore ├── README.md ├── cmake │ ├── CMakeLists.txt │ ├── MLIR.cmake │ └── mhlo.cmake ├── dialects │ ├── CMakeLists.txt │ ├── include │ │ ├── CMakeLists.txt │ │ └── byteir │ │ │ ├── CMakeLists.txt │ │ │ └── Dialect │ │ │ ├── Ace │ │ │ ├── AceBase.td │ │ │ ├── AceDialect.h │ │ │ ├── AceOps.td │ │ │ └── CMakeLists.txt │ │ │ ├── CMakeLists.txt │ │ │ └── Ccl │ │ │ ├── CMakeLists.txt │ │ │ └── IR │ │ │ ├── CMakeLists.txt │ │ │ ├── CclBase.td │ │ │ ├── CclOpInterface.td │ │ │ ├── CclOps.h │ │ │ └── CclOps.td │ └── lib │ │ ├── CMakeLists.txt │ │ └── Dialect │ │ ├── Ace │ │ ├── CMakeLists.txt │ │ └── IR │ │ │ └── AceDialect.cpp │ │ ├── CMakeLists.txt │ │ └── Ccl │ │ ├── CMakeLists.txt │ │ └── IR │ │ ├── CMakeLists.txt │ │ └── CclOps.cpp ├── doc │ ├── attention.md │ ├── byteir_hlo_custom_call.md │ ├── codegen.md │ ├── gpu.md │ ├── linalg.md │ ├── passes.md │ └── rng.md ├── include │ ├── CMakeLists.txt │ ├── byteir-c │ │ ├── Dialects.h │ │ ├── PDLValue.h │ │ ├── Passes.h │ │ └── Translation.h │ └── byteir │ │ ├── Analysis │ │ ├── Alias.h │ │ ├── DimFlag.h │ │ ├── Liveness.h │ │ ├── OpDependence.h │ │ ├── ShapeAnalysis.h │ │ ├── SideEffect.h │ │ ├── SymbolicShape.h │ │ └── UseRange.h │ │ ├── CMakeLists.txt │ │ ├── Conversion │ │ ├── CMakeLists.txt │ │ ├── Common │ │ │ └── FunctionSupport.h │ │ ├── FuncToByre │ │ │ └── FuncToByre.h │ │ ├── GPUToNVVM │ │ │ └── GPUToNVVM.h │ │ ├── HloToByreTensor │ │ │ ├── HloToByreCustom.h │ │ │ └── HloToByreTensor.h │ │ ├── HloToCat │ │ │ ├── ConvertHloToCat.h │ │ │ ├── FuseHloToCat.h │ │ │ └── HloToCat.h │ │ ├── HloToTensor │ │ │ └── ConvertHloToTensor.h │ │ ├── LcclToByre │ │ │ └── LcclToByre.h │ │ ├── MemrefToByre │ │ │ └── MemrefToByre.h │ │ ├── Passes.h │ │ ├── Passes.td │ │ ├── ToAIT │ │ │ └── ToAIT.h │ │ ├── ToAce │ │ │ └── MhloToAce.h │ │ ├── ToByre │ │ │ └── ToByre.h │ │ ├── ToGPU │ │ │ ├── ToGPU.h │ │ │ └── Utils.h │ │ ├── ToHlo │ │ │ └── ArithToMhlo.h │ │ ├── ToLLVM │ │ │ └── ToLLVM.h │ │ ├── ToLinalg │ │ │ └── ToLinalg.h │ │ ├── ToPTX │ │ │ └── ToPTX.h │ │ └── ToTIT │ │ │ └── ToTIT.h │ │ ├── Dialect │ │ ├── Ace │ │ │ ├── CMakeLists.txt │ │ │ ├── Passes.h │ │ │ ├── Passes.td │ │ │ └── Transforms │ │ │ │ └── BufferizableOpInterfaceImpl.h │ │ ├── Affine │ │ │ ├── CMakeLists.txt │ │ │ ├── Passes.h │ │ │ ├── Passes.td │ │ │ └── Transforms │ │ │ │ ├── AffineLoopFusionEx.h │ │ │ │ ├── InsertTrivialAffineLoop.h │ │ │ │ └── RewriteAffineToMemref.h │ │ ├── Byre │ │ │ ├── ByreBase.td │ │ │ ├── ByreDialect.h │ │ │ ├── ByreOps.td │ │ │ ├── CMakeLists.txt │ │ │ ├── Common.h │ │ │ ├── Passes.h │ │ │ ├── Passes.td │ │ │ ├── Serialization.h │ │ │ ├── Serialization │ │ │ │ ├── ByreSerial.td │ │ │ │ ├── ByreSerialOps.h │ │ │ │ ├── CMakeLists.txt │ │ │ │ └── Versioning.h │ │ │ └── Transforms │ │ │ │ ├── BufferizableOpInterfaceImpl.h │ │ │ │ └── Serial.h │ │ ├── CMakeLists.txt │ │ ├── Cat │ │ │ ├── CMakeLists.txt │ │ │ └── IR │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── CatBase.td │ │ │ │ ├── CatDialect.h │ │ │ │ └── CatOps.td │ │ ├── Ccl │ │ │ ├── CMakeLists.txt │ │ │ ├── Passes.h │ │ │ ├── Passes.td │ │ │ ├── TransformOps │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── CclTransformOps.h │ │ │ │ └── CclTransformOps.td │ │ │ └── Transforms │ │ │ │ ├── CclBufferizeOpInterfaceImpl.h │ │ │ │ └── CclMoveDown.h │ │ ├── GPU │ │ │ ├── CMakeLists.txt │ │ │ ├── Passes.h │ │ │ ├── Passes.td │ │ │ ├── TransformOps │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── GPUExtTransformOps.h │ │ │ │ ├── GPUExtTransformOps.td │ │ │ │ └── Utils.h │ │ │ └── Transforms │ │ │ │ ├── GPUBlockSwizzle.h │ │ │ │ ├── GPUDistributeSharedMemoryCopy.h │ │ │ │ ├── GPUDistributeToWarp.h │ │ │ │ ├── GPUPackSharedMemoryAlloc.h │ │ │ │ ├── GPUTensorCoreVectorization.h │ │ │ │ ├── OptimizeVectorTransfer.h │ │ │ │ ├── RemoveTrivialLoops.h │ │ │ │ ├── Transforms.h │ │ │ │ └── Utils.h │ │ ├── Lace │ │ │ ├── CMakeLists.txt │ │ │ ├── LaceBase.td │ │ │ ├── LaceDialect.h │ │ │ └── LaceOps.td │ │ ├── Lccl │ │ │ ├── CMakeLists.txt │ │ │ ├── LcclBase.td │ │ │ ├── LcclOps.h │ │ │ └── LcclOps.td │ │ ├── Linalg │ │ │ ├── CMakeLists.txt │ │ │ ├── IR │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── LinalgExtBase.td │ │ │ │ ├── LinalgExtInterfaces.h │ │ │ │ ├── LinalgExtInterfaces.td │ │ │ │ ├── LinalgExtOps.h │ │ │ │ └── LinalgExtOps.td │ │ │ ├── Passes.h │ │ │ ├── Passes.td │ │ │ ├── TransformOps │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── LinalgExtTransformOps.h │ │ │ │ └── LinalgExtTransformOps.td │ │ │ ├── Transforms │ │ │ │ ├── BufferizableOpInterfaceImpl.h │ │ │ │ ├── Bufferize.h │ │ │ │ ├── CanonicalizeExt.h │ │ │ │ ├── FuseElementwise.h │ │ │ │ ├── HoistingExt.h │ │ │ │ ├── LinalgCollapseLoops.h │ │ │ │ ├── LinalgDataPlace.h │ │ │ │ ├── LinalgExtToLoops.h │ │ │ │ ├── LinalgPrefetch.h │ │ │ │ ├── LinalgPromotion.h │ │ │ │ ├── Tiling.h │ │ │ │ ├── TilingUtils.h │ │ │ │ └── Transforms.h │ │ │ └── Util │ │ │ │ └── Util.h │ │ ├── MemRef │ │ │ ├── CMakeLists.txt │ │ │ ├── Passes.h │ │ │ ├── Passes.td │ │ │ ├── Transforms │ │ │ │ ├── ApplyMemRefAffineLayout.h │ │ │ │ ├── ExtractAddressComputation.h │ │ │ │ ├── RemoveCopy.h │ │ │ │ ├── SimplifyLinearizedIndex.h │ │ │ │ └── SimplifyView.h │ │ │ └── Utils │ │ │ │ ├── Layout.h │ │ │ │ ├── MemEffect.h │ │ │ │ └── Ops.h │ │ ├── SCF │ │ │ ├── CMakeLists.txt │ │ │ ├── Passes.h │ │ │ ├── Passes.td │ │ │ ├── Transforms │ │ │ │ ├── ForallCollapsing.h │ │ │ │ ├── FuseNestedForall.h │ │ │ │ ├── InsertTrivialSCFLoop.h │ │ │ │ ├── RemoveSingleIterationLoop.h │ │ │ │ └── TilingInterfaceToSCFFor.h │ │ │ └── Util │ │ │ │ └── Util.h │ │ ├── Shape │ │ │ ├── CMakeLists.txt │ │ │ ├── IR │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── ShapeExtBase.td │ │ │ │ ├── ShapeExtOps.h │ │ │ │ └── ShapeExtOps.td │ │ │ ├── Passes.h │ │ │ ├── Passes.td │ │ │ └── Transforms │ │ │ │ ├── InsertInputShapeConstraint.h │ │ │ │ ├── InsertTieShape.h │ │ │ │ ├── ResolveShapeConstraint.h │ │ │ │ └── SetAssumingAlwaysTrue.h │ │ ├── Tensor │ │ │ ├── CMakeLists.txt │ │ │ ├── IR │ │ │ │ └── TilingInterfaceImpl.h │ │ │ ├── Passes.h │ │ │ ├── Passes.td │ │ │ └── Transforms │ │ │ │ ├── CanonicalizeExt.h │ │ │ │ ├── ExtractSliceSpecialization.h │ │ │ │ └── TensorPadSpecialization.h │ │ ├── Transform │ │ │ ├── CMakeLists.txt │ │ │ ├── IR │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── TransformExtOps.h │ │ │ │ └── TransformExtOps.td │ │ │ ├── Passes.h │ │ │ ├── Passes.td │ │ │ └── Transforms │ │ │ │ ├── TransformDialectInterpreter.h │ │ │ │ └── TransformInsertion.h │ │ ├── Vector │ │ │ ├── CMakeLists.txt │ │ │ └── Transforms │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── CanonicalizeExt.h │ │ │ │ ├── MoveForallRegionIntoWarpOp.h │ │ │ │ ├── Passes.h │ │ │ │ ├── Passes.td │ │ │ │ └── VectorWarpDistribute.h │ │ └── mhlo │ │ │ ├── Analysis │ │ │ ├── DimFromBroadcast.h │ │ │ └── ShapeAnalysis.h │ │ │ ├── CMakeLists.txt │ │ │ ├── DynamicShapeOpRegister │ │ │ └── Register.h │ │ │ ├── Passes.h │ │ │ ├── Passes.td │ │ │ ├── Transforms │ │ │ ├── BoundedShapeInference.h │ │ │ ├── CanonicalizeExt.h │ │ │ ├── ClusterConstraint.h │ │ │ ├── ConvertFuncToCustomCall.h │ │ │ ├── ConvertInsertion.h │ │ │ ├── ConvertOpToCustomCall.h │ │ │ ├── DTypeConversion.h │ │ │ ├── DecomposeMhloCustomCallOps.h │ │ │ ├── DynamicShapeClustering.h │ │ │ ├── FuncArgRearrangement.h │ │ │ ├── FuseBMMDimension.h │ │ │ ├── FusionOutlining.h │ │ │ ├── GenericFusionCommon.h │ │ │ ├── HloAggressiveFusion.h │ │ │ ├── HloFolder.h │ │ │ ├── HloFuser.h │ │ │ ├── HloMove.h │ │ │ ├── HloSimplify.h │ │ │ ├── InsertShapeConstraint.h │ │ │ ├── LayoutTransformation.h │ │ │ ├── MatmulLayoutTransform.h │ │ │ ├── MoveCommon.h │ │ │ ├── RewriteWithConstraint.h │ │ │ ├── ShapeReification.h │ │ │ ├── StaticShapeInference.h │ │ │ └── UnfuseBatchNorm.h │ │ │ └── Util │ │ │ ├── CustomCallUtil.h │ │ │ ├── FusionUtil.h │ │ │ ├── ShapeInferUtil.h │ │ │ └── Util.h │ │ ├── Pipelines │ │ ├── AffineOpt.h │ │ ├── AllOpt.h │ │ ├── BufferizeOpt.h │ │ ├── ByreHost.h │ │ ├── ByreOpt.h │ │ ├── ByreTensorOpt.h │ │ ├── CatFusionOpt.h │ │ ├── CatPreprocess.h │ │ ├── Common │ │ │ └── Utils.h │ │ ├── GPU │ │ │ ├── ElementwiseCodegen.h │ │ │ ├── GPUOpt.h │ │ │ ├── LinalgMemrefGPU.h │ │ │ ├── MappingForall.h │ │ │ ├── NVVMCodegen.h │ │ │ └── ReductionCodegen.h │ │ ├── HloFusionOpt.h │ │ ├── HloGraphOpt.h │ │ ├── Host │ │ │ ├── Codegen.h │ │ │ ├── HostOpt.h │ │ │ └── ToLLVM.h │ │ ├── InitAllPipelines.h │ │ ├── LinalgMemrefOpt.h │ │ ├── LinalgTensorOpt.h │ │ ├── SCFOpt.h │ │ └── ShapeOpt.h │ │ ├── Stat │ │ ├── AllocCnt │ │ │ └── AllocCnt.h │ │ ├── Common │ │ │ └── Reg.h │ │ ├── InitAllStats.h │ │ └── OpCnt │ │ │ └── OpCnt.h │ │ ├── Target │ │ ├── CUDA │ │ │ ├── CUDAEmitter.h │ │ │ └── ToCUDA.h │ │ ├── Common │ │ │ ├── Common.h │ │ │ └── EmitUtil.h │ │ ├── Cpp │ │ │ ├── CppEmitter.h │ │ │ └── ToCpp.h │ │ ├── LLVM │ │ │ └── ToLLVMBC.h │ │ └── PTX │ │ │ ├── Passes.h │ │ │ └── ToPTX.h │ │ ├── Transforms │ │ ├── AnchoredPipeline.h │ │ ├── ApplyPDLPatterns.h │ │ ├── Bufferize.h │ │ ├── CMAE.h │ │ ├── CMakeLists.txt │ │ ├── CanonicalizeExt.h │ │ ├── CollectFunc.h │ │ ├── CondCanonicalize.h │ │ ├── FuncTag.h │ │ ├── GenericDeviceConfig.h │ │ ├── GraphClusteringAlgo.h │ │ ├── GraphClusteringByDevice.h │ │ ├── InsertUniqueId.h │ │ ├── LoopTag.h │ │ ├── LoopUnroll.h │ │ ├── MemoryPlanning.h │ │ ├── ModuleTag.h │ │ ├── Passes.h │ │ ├── Passes.td │ │ ├── RemoveFuncBody.h │ │ ├── RewriteOpToStdCall.h │ │ ├── SetArgShape.h │ │ ├── SetSpace.h │ │ ├── ShapeFuncOutlining.h │ │ ├── TryCatchModulePipeline.h │ │ └── Utils.h │ │ └── Utils │ │ ├── AffineUtils.h │ │ ├── AttrUtils.h │ │ ├── FuncUtils.h │ │ ├── GraphUtils.h │ │ ├── HashUtils.h │ │ ├── Hoist.h │ │ ├── IRRewrite.h │ │ ├── LoopUtils.h │ │ ├── MemUtils.h │ │ ├── ModuleUtils.h │ │ ├── OpInterfaceUtils.h │ │ ├── OptionUtils.h │ │ ├── PatternMatch.h │ │ ├── PipelineUtils.h │ │ ├── TileUtils.h │ │ ├── TypeUtils.h │ │ └── Utils.h ├── lib │ ├── Analysis │ │ ├── CMakeLists.txt │ │ ├── DimFlag.cpp │ │ ├── Liveness.cpp │ │ ├── OpDependence.cpp │ │ ├── ShapeAnalysis.cpp │ │ ├── SideEffect.cpp │ │ ├── SymbolicShape.cpp │ │ └── UseRange.cpp │ ├── CAPI │ │ ├── CMakeLists.txt │ │ ├── Dialects.cpp │ │ ├── PDLValue.cpp │ │ ├── Passes.cpp │ │ └── Translation.cpp │ ├── CMakeLists.txt │ ├── Conversion │ │ ├── CMakeLists.txt │ │ ├── Common │ │ │ ├── CMakeLists.txt │ │ │ └── FunctionSupport.cpp │ │ ├── FuncToByre │ │ │ ├── CMakeLists.txt │ │ │ └── FuncToByre.cpp │ │ ├── GPUToNVVM │ │ │ ├── CMakeLists.txt │ │ │ └── GPUToNVVM.cpp │ │ ├── HloToByreTensor │ │ │ ├── CMakeLists.txt │ │ │ ├── HloToByreCustom.cpp │ │ │ └── HloToByreTensor.cpp │ │ ├── HloToCat │ │ │ ├── CMakeLists.txt │ │ │ ├── ConvertHloToCat.cpp │ │ │ ├── FuseHloToCat.cpp │ │ │ ├── FuseHloToCatPattern.td │ │ │ ├── HloToCat.cpp │ │ │ └── Utils.h │ │ ├── HloToTensor │ │ │ ├── CMakeLists.txt │ │ │ └── ConvertHloToTensor.cpp │ │ ├── LcclToByre │ │ │ ├── CMakeLists.txt │ │ │ └── LcclToByre.cpp │ │ ├── MemrefToByre │ │ │ ├── CMakeLists.txt │ │ │ └── MemrefToByre.cpp │ │ ├── PassDetail.h │ │ ├── ToAIT │ │ │ ├── CMakeLists.txt │ │ │ └── GenAITConfig.cpp │ │ ├── ToAce │ │ │ ├── CMakeLists.txt │ │ │ ├── MhloToAce.cpp │ │ │ └── MhloToAceActivationPattern.td │ │ ├── ToByre │ │ │ ├── CMakeLists.txt │ │ │ └── ToByre.cpp │ │ ├── ToGPU │ │ │ ├── CMakeLists.txt │ │ │ ├── CoalescedForToGPU.cpp │ │ │ ├── FuncToGPU.cpp │ │ │ └── Utils.cpp │ │ ├── ToHlo │ │ │ ├── ArithToMhlo.cpp │ │ │ ├── ArithToMhloPattern.td │ │ │ └── CMakeLists.txt │ │ ├── ToLLVM │ │ │ ├── CMakeLists.txt │ │ │ ├── CollectFuncToLLVM.cpp │ │ │ └── GenLLVMConfig.cpp │ │ ├── ToLinalg │ │ │ ├── CMakeLists.txt │ │ │ ├── HloToLinalg.cpp │ │ │ ├── LinalgExtToLinalg.cpp │ │ │ ├── MemrefCopyToLinalg.cpp │ │ │ ├── TensorToLinalg.cpp │ │ │ └── UnrealizedCastToLinalg.cpp │ │ ├── ToPTX │ │ │ ├── CMakeLists.txt │ │ │ ├── CollectGPUKernel.cpp │ │ │ └── GenPTXConfig.cpp │ │ └── ToTIT │ │ │ ├── CMakeLists.txt │ │ │ └── GenTITConfig.cpp │ ├── Dialect │ │ ├── Ace │ │ │ ├── CMakeLists.txt │ │ │ └── Transforms │ │ │ │ ├── BufferizableOpInterfaceImpl.cpp │ │ │ │ ├── Bufferize.cpp │ │ │ │ └── PassDetail.h │ │ ├── Affine │ │ │ ├── CMakeLists.txt │ │ │ └── Transforms │ │ │ │ ├── AffineLoopFusionEx.cpp │ │ │ │ ├── InsertTrivialAffineLoop.cpp │ │ │ │ ├── PassDetail.h │ │ │ │ └── RewriteAffineToMemref.cpp │ │ ├── Byre │ │ │ ├── CMakeLists.txt │ │ │ ├── IR │ │ │ │ ├── ByreDialect.cpp │ │ │ │ ├── Common.cpp │ │ │ │ ├── Serialization.cpp │ │ │ │ └── Serialization │ │ │ │ │ ├── ByreSerialOps.cpp │ │ │ │ │ ├── Bytecode.cpp │ │ │ │ │ ├── Bytecode.h │ │ │ │ │ └── Versioning.cpp │ │ │ └── Transforms │ │ │ │ ├── BufferizableOpInterfaceImpl.cpp │ │ │ │ ├── PassDetail.h │ │ │ │ └── Serial.cpp │ │ ├── CMakeLists.txt │ │ ├── Cat │ │ │ ├── CMakeLists.txt │ │ │ └── IR │ │ │ │ ├── CMakeLists.txt │ │ │ │ └── CatDialect.cpp │ │ ├── Ccl │ │ │ ├── CMakeLists.txt │ │ │ ├── TransformOps │ │ │ │ ├── CMakeLists.txt │ │ │ │ └── CclTransformOps.cpp │ │ │ └── Transforms │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── CclBufferizeOpInterfaceImpl.cpp │ │ │ │ ├── CclMoveDown.cpp │ │ │ │ └── PassDetail.h │ │ ├── GPU │ │ │ ├── CMakeLists.txt │ │ │ ├── TransformOps │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── GPUExtTransformOps.cpp │ │ │ │ └── Utils.cpp │ │ │ └── Transforms │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── GPUBlockSwizzle.cpp │ │ │ │ ├── GPUDistributeSharedMemoryCopy.cpp │ │ │ │ ├── GPUDistributeToWarp.cpp │ │ │ │ ├── GPUPackSharedMemoryAlloc.cpp │ │ │ │ ├── GPUTensorCoreVectorization.cpp │ │ │ │ ├── OptimizeVectorTransfer.cpp │ │ │ │ ├── PassDetail.h │ │ │ │ ├── RemoveTrivialLoops.cpp │ │ │ │ ├── ShmAllocaToWorkgroupArg.cpp │ │ │ │ └── Utils.cpp │ │ ├── Lace │ │ │ ├── CMakeLists.txt │ │ │ └── IR │ │ │ │ └── LaceDialect.cpp │ │ ├── Lccl │ │ │ ├── CMakeLists.txt │ │ │ └── IR │ │ │ │ ├── CMakeLists.txt │ │ │ │ └── LcclOps.cpp │ │ ├── Linalg │ │ │ ├── CMakeLists.txt │ │ │ ├── IR │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── LinalgExtInterfaces.cpp │ │ │ │ └── LinalgExtOps.cpp │ │ │ ├── TransformOps │ │ │ │ ├── CMakeLists.txt │ │ │ │ └── LinalgExtTransformOps.cpp │ │ │ ├── Transforms │ │ │ │ ├── BufferizableOpInterfaceImpl.cpp │ │ │ │ ├── Bufferize.cpp │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── CanonicalizeExt.cpp │ │ │ │ ├── FuseElementwise.cpp │ │ │ │ ├── HoistingExt.cpp │ │ │ │ ├── LinalgCollapseLoops.cpp │ │ │ │ ├── LinalgDataPlace.cpp │ │ │ │ ├── LinalgExtToLoops.cpp │ │ │ │ ├── LinalgGeneralizationExt.cpp │ │ │ │ ├── LinalgPrefetch.cpp │ │ │ │ ├── LinalgPromotion.cpp │ │ │ │ ├── PassDetail.h │ │ │ │ ├── ScopeTiling.cpp │ │ │ │ ├── Tiling.cpp │ │ │ │ ├── TilingUtils.cpp │ │ │ │ └── Transforms.cpp │ │ │ └── Util │ │ │ │ ├── CMakeLists.txt │ │ │ │ └── Util.cpp │ │ ├── MemRef │ │ │ ├── CMakeLists.txt │ │ │ ├── Transforms │ │ │ │ ├── ApplyMemRefAffineLayout.cpp │ │ │ │ ├── ExtractAddressComputation.cpp │ │ │ │ ├── PassDetail.h │ │ │ │ ├── RemoveCopy.cpp │ │ │ │ ├── SimplifyLinearizedIndex.cpp │ │ │ │ └── SimplifyView.cpp │ │ │ └── Utils │ │ │ │ ├── Layout.cpp │ │ │ │ ├── MemEffect.cpp │ │ │ │ └── Ops.cpp │ │ ├── SCF │ │ │ ├── CMakeLists.txt │ │ │ ├── Transforms │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── ForallCollapsing.cpp │ │ │ │ ├── FuseNestedForall.cpp │ │ │ │ ├── InsertTrivialSCFLoop.cpp │ │ │ │ ├── PassDetail.h │ │ │ │ ├── RemoveSingleIterationLoop.cpp │ │ │ │ └── TilingInterfaceToSCFFor.cpp │ │ │ └── Util │ │ │ │ ├── CMakeLists.txt │ │ │ │ └── Util.cpp │ │ ├── Shape │ │ │ ├── CMakeLists.txt │ │ │ ├── IR │ │ │ │ └── ShapeExtOps.cpp │ │ │ └── Transforms │ │ │ │ ├── InsertInputShapeConstraint.cpp │ │ │ │ ├── InsertTieShape.cpp │ │ │ │ ├── PassDetail.h │ │ │ │ ├── ResolveShapeConstraint.cpp │ │ │ │ └── SetAssumingAlwaysTrue.cpp │ │ ├── Tensor │ │ │ ├── CMakeLists.txt │ │ │ ├── IR │ │ │ │ ├── CMakeLists.txt │ │ │ │ └── TilingInterfaceImpl.cpp │ │ │ └── Transforms │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── CanonicalizeExt.cpp │ │ │ │ ├── ExtractSliceSpecialization.cpp │ │ │ │ ├── PassDetail.h │ │ │ │ └── TensorPadSpecialization.cpp │ │ ├── Transform │ │ │ ├── CMakeLists.txt │ │ │ ├── IR │ │ │ │ ├── CMakeLists.txt │ │ │ │ └── TransformExtOps.cpp │ │ │ └── Transforms │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── PassDetail.h │ │ │ │ ├── TransformDialectInterpreter.cpp │ │ │ │ └── TransformInsertion.cpp │ │ ├── Vector │ │ │ ├── CMakeLists.txt │ │ │ └── Transforms │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── CanonicalizeExt.cpp │ │ │ │ ├── MoveForallRegionIntoWarpOp.cpp │ │ │ │ ├── PassDetail.h │ │ │ │ ├── VectorLowerings.cpp │ │ │ │ └── VectorWarpDistribute.cpp │ │ └── mhlo │ │ │ ├── Analysis │ │ │ ├── DimFromBroadcast.cpp │ │ │ └── ShapeAnalysis.cpp │ │ │ ├── CMakeLists.txt │ │ │ ├── DynamicShapeOpRegister │ │ │ ├── AddN.cpp │ │ │ ├── BatchMatMul.cpp │ │ │ ├── Concatenate.cpp │ │ │ ├── Convolution.cpp │ │ │ ├── DotLike.cpp │ │ │ ├── DynamicBroadcastInDim.cpp │ │ │ ├── DynamicPartition.cpp │ │ │ ├── DynamicStitchLike.cpp │ │ │ ├── Einsum.cpp │ │ │ ├── GeLU.cpp │ │ │ ├── LayerNorm.cpp │ │ │ ├── NonZero.cpp │ │ │ ├── OneHot.cpp │ │ │ ├── RealDynamicSlice.cpp │ │ │ ├── Reduce.cpp │ │ │ ├── Repeat.cpp │ │ │ ├── ReshapeLike.cpp │ │ │ ├── ScatterNd.cpp │ │ │ ├── Softmax.cpp │ │ │ ├── StridedSlice.cpp │ │ │ └── TorchIndexSelect.cpp │ │ │ ├── Transforms │ │ │ ├── BoundedShapeInference.cpp │ │ │ ├── CanonicalizeExt.cpp │ │ │ ├── ClusterConstraint.cpp │ │ │ ├── ConvBackwardFusion.cpp │ │ │ ├── ConvForwardFusion.cpp │ │ │ ├── ConvertFuncToCustomCall.cpp │ │ │ ├── ConvertInsertion.cpp │ │ │ ├── ConvertOpToCustomCall.cpp │ │ │ ├── DTypeConversion.cpp │ │ │ ├── DecomposeMhloCustomCallOps.cpp │ │ │ ├── DynamicShapeClustering.cpp │ │ │ ├── FuncArgRearrangement.cpp │ │ │ ├── FuseBMMDimension.cpp │ │ │ ├── FuseTransposeIntoDotGeneral.cpp │ │ │ ├── FusionOutlining.cpp │ │ │ ├── GenericFusion.cpp │ │ │ ├── HloFolder.cpp │ │ │ ├── HloMoveDown.cpp │ │ │ ├── HloMoveUp.cpp │ │ │ ├── HloSimplify.cpp │ │ │ ├── IOConvertFusion.cpp │ │ │ ├── InsertShapeConstraint.cpp │ │ │ ├── LayoutTransformation.cpp │ │ │ ├── MatmulLayoutTransform.cpp │ │ │ ├── PassDetail.h │ │ │ ├── ReduceWindowFusion.cpp │ │ │ ├── RewriteWithConstraint.cpp │ │ │ ├── ShapeReification.cpp │ │ │ ├── StaticShapeInference.cpp │ │ │ ├── TrivialFusion.cpp │ │ │ └── UnfuseBatchNorm.cpp │ │ │ └── Util │ │ │ ├── FusionUtil.cpp │ │ │ ├── ShapeInferUtil.cpp │ │ │ └── Util.cpp │ ├── Pipelines │ │ ├── AffineOpt.cpp │ │ ├── AllOpt.cpp │ │ ├── BufferizeOpt.cpp │ │ ├── ByreHost.cpp │ │ ├── ByreOpt.cpp │ │ ├── ByreTensorOpt.cpp │ │ ├── CMakeLists.txt │ │ ├── CatFusionOpt.cpp │ │ ├── CatPreprocess.cpp │ │ ├── Common │ │ │ ├── CMakeLists.txt │ │ │ └── Utils.cpp │ │ ├── GPU │ │ │ ├── CMakeLists.txt │ │ │ ├── ElementwiseCodegen.cpp │ │ │ ├── GPUOpt.cpp │ │ │ ├── LinalgMemrefGPU.cpp │ │ │ ├── MappingForall.cpp │ │ │ ├── NVVMCodegen.cpp │ │ │ └── ReductionCodegen.cpp │ │ ├── HloFusionOpt.cpp │ │ ├── HloGraphOpt.cpp │ │ ├── Host │ │ │ ├── CMakeLists.txt │ │ │ ├── Codegen.cpp │ │ │ ├── HostOpt.cpp │ │ │ └── ToLLVM.cpp │ │ ├── LinalgMemrefOpt.cpp │ │ ├── LinalgTensorOpt.cpp │ │ ├── SCFOpt.cpp │ │ └── ShapeOpt.cpp │ ├── Stat │ │ ├── AllocCnt │ │ │ ├── AllocCnt.cpp │ │ │ └── CMakeLists.txt │ │ ├── CMakeLists.txt │ │ ├── Common │ │ │ ├── CMakeLists.txt │ │ │ └── Reg.cpp │ │ └── OpCnt │ │ │ ├── CMakeLists.txt │ │ │ └── OpCnt.cpp │ ├── Target │ │ ├── CMakeLists.txt │ │ ├── CUDA │ │ │ ├── CMakeLists.txt │ │ │ ├── TranslateRegistration.cpp │ │ │ └── TranslateToCUDA.cpp │ │ ├── Cpp │ │ │ ├── CMakeLists.txt │ │ │ ├── TranslateRegistration.cpp │ │ │ └── TranslateToCpp.cpp │ │ ├── LLVM │ │ │ ├── CMakeLists.txt │ │ │ └── TranslateRegistration.cpp │ │ └── PTX │ │ │ ├── CMakeLists.txt │ │ │ ├── GPUKernelToPTX.cpp │ │ │ ├── TranslateRegistration.cpp │ │ │ └── TranslateToPTX.cpp │ ├── Transforms │ │ ├── AnchoredPipeline.cpp │ │ ├── ApplyPDLPatterns.cpp │ │ ├── Bufferize.cpp │ │ ├── CMAE.cpp │ │ ├── CMakeLists.txt │ │ ├── CanonicalizeExt.cpp │ │ ├── CollectFunc.cpp │ │ ├── CondCanonicalize.cpp │ │ ├── FuncTag.cpp │ │ ├── GenericDeviceConfig.cpp │ │ ├── GraphClusteringByDevice.cpp │ │ ├── InsertUniqueId.cpp │ │ ├── LoopTag.cpp │ │ ├── LoopUnroll.cpp │ │ ├── MemoryPlanning.cpp │ │ ├── ModuleTag.cpp │ │ ├── PassDetail.h │ │ ├── RemoveFuncBody.cpp │ │ ├── RewriteOpToStdCall.cpp │ │ ├── SetArgShape.cpp │ │ ├── SetSpace.cpp │ │ ├── ShapeFuncOutlining.cpp │ │ ├── TryCatchModulePipeline.cpp │ │ └── Utils.cpp │ └── Utils │ │ ├── AffineUtils.cpp │ │ ├── AttrUtils.cpp │ │ ├── CMakeLists.txt │ │ ├── FuncUtils.cpp │ │ ├── GraphUtils.cpp │ │ ├── Hoist.cpp │ │ ├── IRRewrite.cpp │ │ ├── LoopUtils.cpp │ │ ├── MemUtils.cpp │ │ ├── ModuleUtils.cpp │ │ ├── OpInterfaceUtils.cpp │ │ ├── OptionUtils.cpp │ │ ├── PatternMatch.cpp │ │ ├── PipelineUtils.cpp │ │ ├── TileUtils.cpp │ │ ├── TypeUtils.cpp │ │ └── Utils.cpp ├── numerical │ ├── CMakeLists.txt │ ├── hlo │ │ ├── canonicalize_ext.mlir │ │ ├── conv_bn.mlir │ │ ├── dot_bn.mlir │ │ ├── hlo_fold.mlir │ │ ├── hlo_move_down.mlir │ │ ├── hlo_simplify.mlir │ │ ├── numerical_test.py │ │ ├── slice_move_down_and_merge.mlir │ │ └── test_broadcast_dense_elements_attr.mlir │ ├── lit.cfg.py │ └── lit.site.cfg.py.in ├── python │ ├── ByteIRModules.cpp │ ├── CMakeLists.txt │ ├── byteir │ │ ├── README.md │ │ ├── __init__.py │ │ ├── _backend_registry.py │ │ ├── _mlir_libs │ │ │ └── _site_initialize_0.py │ │ ├── compile.py │ │ ├── dialects │ │ │ ├── CatOps.td │ │ │ └── cat │ │ │ │ ├── __init__.py │ │ │ │ ├── ait_cache.py │ │ │ │ ├── ir_processor.py │ │ │ │ ├── ir_translator │ │ │ │ ├── __init__.py │ │ │ │ ├── ait_builder.py │ │ │ │ ├── backend │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── ait_registry.py │ │ │ │ │ └── tit_registry.py │ │ │ │ ├── tit_builder.py │ │ │ │ └── translator.py │ │ │ │ └── tit_cache.py │ │ ├── pattern_matches.py │ │ ├── tools │ │ │ ├── cat_executor.py │ │ │ └── compiler.py │ │ └── utils.py │ ├── gen_version.py │ ├── setup.py │ ├── test │ │ ├── CMakeLists.txt │ │ ├── api │ │ │ ├── test_pattern_matches.py │ │ │ └── test_py_api.py │ │ ├── dialects │ │ │ └── cat │ │ │ │ └── ait │ │ │ │ ├── numerical │ │ │ │ ├── layernorm.mlir │ │ │ │ ├── matmul.mlir │ │ │ │ ├── permute021.mlir │ │ │ │ ├── permute0213.mlir │ │ │ │ ├── permute0312.mlir │ │ │ │ ├── permute10.mlir │ │ │ │ └── softmax_f16.mlir │ │ │ │ └── profile │ │ │ │ └── matmul.mlir │ │ ├── lit.cfg.py │ │ └── lit.site.cfg.py.in │ └── version.txt ├── scripts │ ├── README.md │ ├── gen_testcases.py │ ├── gen_testcases_and_check_diff.sh │ └── sync_to_runtime.sh ├── test │ ├── Analysis │ │ ├── testPrintArgSideEffect.mlir │ │ ├── testPrintLiveness.mlir │ │ ├── testPrintShapeAnalysis.mlir │ │ ├── testPrintSymbolicShape.mlir │ │ └── testPrintUseRange.mlir │ ├── CMakeLists.txt │ ├── CPURunner │ │ ├── gelu.mlir │ │ ├── repeatCustomCall.mlir │ │ └── scatterTiling.mlir │ ├── Conversion │ │ ├── FuncToByre │ │ │ └── func_to_byre_tensor.mlir │ │ ├── HloToByreTensor │ │ │ └── compute_ops.mlir │ │ ├── HloToCat │ │ │ ├── basic_ops.mlir │ │ │ └── fused_ops.mlir │ │ ├── HloToTensor │ │ │ └── scatter_to_insertslice.mlir │ │ ├── LcclToByre │ │ │ └── lcclToByre.mlir │ │ ├── MemrefToByre │ │ │ └── memref_to_byre.mlir │ │ ├── ToAce │ │ │ └── mhloToAceActivation.mlir │ │ ├── ToByre │ │ │ ├── convertFuncAndCallToByre.mlir │ │ │ └── convertMemRefToByre.mlir │ │ ├── ToCUDAGPU │ │ │ ├── fusionGPUToNVVM.mlir │ │ │ ├── fusionGPUToNVVMBarePtr.mlir │ │ │ └── fusionHloToGPU.mlir │ │ ├── ToGPU │ │ │ ├── coalescedForToGPU.mlir │ │ │ └── funcToGPU.mlir │ │ ├── ToHlo │ │ │ └── arithConstToMhlo.mlir │ │ ├── ToLinalg │ │ │ ├── LinalgExtToLinalg.mlir │ │ │ ├── TesnorToLinalg.mlir │ │ │ ├── fusionHlo.mlir │ │ │ ├── hloConvertToLinalg.mlir │ │ │ ├── memrefcopyToLinalg.mlir │ │ │ ├── primitiveOpsHlo.mlir │ │ │ ├── reducef16.mlir │ │ │ ├── repeatCustomCallToLinalg.mlir │ │ │ ├── rngCustomCallToLinalg.mlir │ │ │ ├── simpleHlo.mlir │ │ │ └── unrealizedCastToLinalg.mlir │ │ ├── ToPTX │ │ │ ├── genPTXConfig.mlir │ │ │ └── genPTXConfigBarePtr.mlir │ │ └── VectorToGPU │ │ │ └── existing-vector-to-mma-ops.mlir │ ├── Dialect │ │ ├── Ace │ │ │ ├── attrs.mlir │ │ │ ├── bufferize.mlir │ │ │ ├── canonicalize.mlir │ │ │ └── ops.mlir │ │ ├── Affine │ │ │ ├── affineLoopFusionEx.mlir │ │ │ ├── affineToMemRef.mlir │ │ │ └── insertTrivialAffineLoop.mlir │ │ ├── Byre │ │ │ ├── Serialization │ │ │ │ ├── Compatibility │ │ │ │ │ ├── version_1_0_0.mlir │ │ │ │ │ ├── version_1_0_0.mlir.bc │ │ │ │ │ ├── version_1_0_0.mlir.bc.v0 │ │ │ │ │ ├── version_1_0_0_alloc.mlir │ │ │ │ │ └── version_1_0_0_alloc.mlir.bc │ │ │ │ └── round_trip.mlir │ │ │ ├── bert_transformer.mlir │ │ │ ├── buffer_ops.mlir │ │ │ ├── bufferize.mlir │ │ │ ├── canonicalize.mlir │ │ │ ├── interface.mlir │ │ │ └── invalid.mlir │ │ ├── Cat │ │ │ └── ops.mlir │ │ ├── Ccl │ │ │ ├── ccl_bufferize.mlir │ │ │ ├── ccl_canonicalize.mlir │ │ │ ├── ccl_move_down.mlir │ │ │ ├── decompose_all_reduce.mlir │ │ │ ├── invalid.mlir │ │ │ └── ops.mlir │ │ ├── GPU │ │ │ ├── gpu-block-swizzle.mlir │ │ │ ├── gpu-distribute-shared-memory-copy.mlir │ │ │ ├── gpu-distributed-to-warp.mlir │ │ │ ├── gpu-pack-shared-memory-alloc.mlir │ │ │ ├── gpu-tensorcore-vectorization.mlir │ │ │ ├── optimize-vector-transfer.mlir │ │ │ ├── remove-trivial-loops.mlir │ │ │ ├── transform-gpu-failing.mlir │ │ │ ├── transform-map-forall-to-blocks.mlir │ │ │ └── transform-map-nested-forall-to-threads.mlir │ │ ├── Lace │ │ │ ├── invalid.mlir │ │ │ └── ops.mlir │ │ ├── Linalg │ │ │ ├── annotate.mlir │ │ │ ├── bufferize.mlir │ │ │ ├── canonicalizeExt.mlir │ │ │ ├── dataPlace-lagacy.mlir │ │ │ ├── dataPlace-tensor.mlir │ │ │ ├── extension.mlir │ │ │ ├── fuse-attention-upstream.mlir │ │ │ ├── fuse-attention.mlir │ │ │ ├── generalization.mlir │ │ │ ├── linalg-collapse-loops.mlir │ │ │ ├── linalg-fuse-elementwise-ext-existing.mlir │ │ │ ├── linalg-fuse-elementwise-ext.mlir │ │ │ ├── linalg-promotion-epilogue-fusion.mlir │ │ │ ├── linalg-promotion.mlir │ │ │ ├── linalgExtToLoops.mlir │ │ │ ├── opTiling1.mlir │ │ │ ├── opTiling2.mlir │ │ │ ├── prefetch.mlir │ │ │ ├── scopeTiling3SplitK-dev.mlir │ │ │ ├── transform-dev.mlir │ │ │ ├── transform-lower-to-loops.mlir │ │ │ ├── transform-op-collapse-dims.mlir │ │ │ ├── transform-op-fold-unit-extent-dims.mlir │ │ │ ├── transform-op-fuse-dev.mlir │ │ │ ├── transform-op-fuse-into-containing.mlir │ │ │ ├── transform-op-fuse-multi-root.mlir │ │ │ ├── transform-op-fuse.mlir │ │ │ ├── transform-op-shared-out-to-dist-style.mlir │ │ │ ├── transform-op-tile-ext.mlir │ │ │ ├── transform-op-tile-loop-hint.mlir │ │ │ └── transform-op-tile-reduction-parallel.mlir │ │ ├── MemRef │ │ │ ├── canonicalize.mlir │ │ │ ├── layout.mlir │ │ │ ├── removeCopy.mlir │ │ │ └── simplifyView.mlir │ │ ├── Mhlo │ │ │ ├── fusion.mlir │ │ │ ├── multi_return.mlir │ │ │ ├── reduce.mlir │ │ │ ├── simple.mlir │ │ │ └── transforms │ │ │ │ ├── ConvBNFolder.mlir │ │ │ │ ├── ConvBackwardFusion.mlir │ │ │ │ ├── ConvBiasActFusion.mlir │ │ │ │ ├── ConvertOpToCustomCall.mlir │ │ │ │ ├── DecomposeMhloCustomCallOps.mlir │ │ │ │ ├── IOConvertFusion.mlir │ │ │ │ ├── LayoutTransformation.mlir │ │ │ │ ├── RewriteWithConstraint.mlir │ │ │ │ ├── SliceMoveDownAndMerge.mlir │ │ │ │ ├── TestBroadcastDenseElementsAttr.mlir │ │ │ │ ├── TestConvertFuncToCustomCall.mlir │ │ │ │ ├── TestConvertInsertion.mlir │ │ │ │ ├── TestCustomConvert.mlir │ │ │ │ ├── TestDTypeConversion.mlir │ │ │ │ ├── TestDTypeConversionModifyFunc.mlir │ │ │ │ ├── TestFuncArgRearrangement.mlir │ │ │ │ ├── aggressiveFusion.mlir │ │ │ │ ├── canonicalize │ │ │ │ ├── arithOptimize.mlir │ │ │ │ ├── dynamicGather.mlir │ │ │ │ └── transposeFolder.mlir │ │ │ │ ├── clusterConstraint.mlir │ │ │ │ ├── concatSliceFusion.mlir │ │ │ │ ├── elementFusion.mlir │ │ │ │ ├── expandHloTuples.mlir │ │ │ │ ├── fuseBMMDimension.mlir │ │ │ │ ├── fuseTransposeIntoDotGeneral.mlir │ │ │ │ ├── fusionOutlining.mlir │ │ │ │ ├── hloFolder.mlir │ │ │ │ ├── hloMoveDown.mlir │ │ │ │ ├── hloMoveUp.mlir │ │ │ │ ├── hloSimplify.mlir │ │ │ │ ├── insertShapeConstraint.mlir │ │ │ │ ├── matmulEpilogueFusion.mlir │ │ │ │ ├── matmulLayoutTransform.mlir │ │ │ │ ├── mhloFlattenTuple.mlir │ │ │ │ └── reduceFusion.mlir │ │ ├── SCF │ │ │ ├── forallCollapsing.mlir │ │ │ ├── fuseNestedForall.mlir │ │ │ ├── insertTrivialSCFLoop.mlir │ │ │ └── moveForallRegionIntoWarpOp.mlir │ │ ├── Shape │ │ │ ├── insertInputShapeConstraint.mlir │ │ │ ├── insertTieShape.mlir │ │ │ ├── resolveShapeConstraint.mlir │ │ │ └── setAssumingAlwaysTrue.mlir │ │ ├── Tensor │ │ │ └── canonicalizeExt.mlir │ │ ├── Transform │ │ │ ├── canonicalize.mlir │ │ │ ├── cleanup.mlir │ │ │ ├── detensorizeInsertion.mlir │ │ │ ├── dump.mlir │ │ │ ├── transformDialectInterpreter.mlir │ │ │ └── transformInsertion.mlir │ │ └── Vector │ │ │ └── canonicalizeExt.mlir │ ├── E2E │ │ ├── CUDA │ │ │ ├── AliasLikeGPU │ │ │ │ ├── 10b_ptx_codegen.mlir │ │ │ │ ├── 1_hlo_opt.mlir │ │ │ │ ├── 2_linalg_tensor_opt.mlir │ │ │ │ ├── 3_byre_tensor_opt.mlir │ │ │ │ ├── 4_bufferize_opt.mlir │ │ │ │ ├── 5_alternative_scf_opt.mlir │ │ │ │ ├── 6_gpu_opt.mlir │ │ │ │ ├── 7_set_space_opt.mlir │ │ │ │ ├── 8_byre_opt.mlir │ │ │ │ ├── 9a_byre_host.mlir │ │ │ │ ├── 9b_nvvm_codegen.mlir │ │ │ │ ├── device_output.ptx │ │ │ │ ├── host_output.mlir │ │ │ │ ├── input.mlir │ │ │ │ └── template.py │ │ │ ├── BertTiny │ │ │ │ ├── BW │ │ │ │ │ └── input.mlir │ │ │ │ └── FW │ │ │ │ │ ├── 1_hlo_opt.mlir │ │ │ │ │ └── input.mlir │ │ │ ├── CclInference │ │ │ │ └── input.mlir │ │ │ ├── MLPBasic │ │ │ │ ├── 1_preprocess_for_lowering.mlir │ │ │ │ └── input.mlir │ │ │ ├── MLPInference │ │ │ │ ├── 10b_ptx_codegen.mlir │ │ │ │ ├── 1_hlo_opt.mlir │ │ │ │ ├── 2_linalg_tensor_opt.mlir │ │ │ │ ├── 3_byre_tensor_opt.mlir │ │ │ │ ├── 4_bufferize_opt.mlir │ │ │ │ ├── 5_affine_opt.mlir │ │ │ │ ├── 5_alternative_scf_opt.mlir │ │ │ │ ├── 6_gpu_opt.mlir │ │ │ │ ├── 7_set_space_opt.mlir │ │ │ │ ├── 8_byre_opt.mlir │ │ │ │ ├── 9a_byre_host.mlir │ │ │ │ ├── 9b_nvvm_codegen.mlir │ │ │ │ ├── device_output.ptx │ │ │ │ ├── host_output.mlir │ │ │ │ ├── input.mlir │ │ │ │ └── template.py │ │ │ ├── NanoGPT │ │ │ │ ├── BW │ │ │ │ │ └── input.mlir │ │ │ │ └── FW │ │ │ │ │ └── input.mlir │ │ │ └── ResNet18 │ │ │ │ ├── BW │ │ │ │ ├── 10b_ptx_codegen.mlir │ │ │ │ ├── 1_hlo_opt.mlir │ │ │ │ ├── 2_linalg_tensor_opt.mlir │ │ │ │ ├── 3_byre_tensor_opt.mlir │ │ │ │ ├── 4_bufferize_opt.mlir │ │ │ │ ├── 5_affine_opt.mlir │ │ │ │ ├── 5_alternative_scf_opt.mlir │ │ │ │ ├── 6_gpu_opt.mlir │ │ │ │ ├── 7_set_space_opt.mlir │ │ │ │ ├── 8_byre_opt.mlir │ │ │ │ ├── 9a_byre_host.mlir │ │ │ │ ├── 9b_nvvm_codegen.mlir │ │ │ │ ├── device_output.ptx │ │ │ │ ├── host_output.mlir │ │ │ │ ├── input.mlir │ │ │ │ └── template.py │ │ │ │ ├── FW │ │ │ │ ├── 10b_ptx_codegen.mlir │ │ │ │ ├── 1_hlo_opt.mlir │ │ │ │ ├── 2_linalg_tensor_opt.mlir │ │ │ │ ├── 3_byre_tensor_opt.mlir │ │ │ │ ├── 4_bufferize_opt.mlir │ │ │ │ ├── 5_affine_opt.mlir │ │ │ │ ├── 5_alternative_scf_opt.mlir │ │ │ │ ├── 6_gpu_opt.mlir │ │ │ │ ├── 7_set_space_opt.mlir │ │ │ │ ├── 8_byre_opt.mlir │ │ │ │ ├── 9a_byre_host.mlir │ │ │ │ ├── 9b_nvvm_codegen.mlir │ │ │ │ ├── device_output.ptx │ │ │ │ ├── host_output.mlir │ │ │ │ ├── input.mlir │ │ │ │ └── template.py │ │ │ │ └── Whole │ │ │ │ ├── 10b_ptx_codegen.mlir │ │ │ │ ├── 1_hlo_opt.mlir │ │ │ │ ├── 2_linalg_tensor_opt.mlir │ │ │ │ ├── 3_byre_tensor_opt.mlir │ │ │ │ ├── 4_bufferize_opt.mlir │ │ │ │ ├── 5_affine_opt.mlir │ │ │ │ ├── 5_alternative_scf_opt.mlir │ │ │ │ ├── 6_gpu_opt.mlir │ │ │ │ ├── 7_set_space_opt.mlir │ │ │ │ ├── 8_byre_opt.mlir │ │ │ │ ├── 9a_byre_host.mlir │ │ │ │ ├── 9b_nvvm_codegen.mlir │ │ │ │ ├── device_output.ptx │ │ │ │ ├── host_output.mlir │ │ │ │ ├── input.mlir │ │ │ │ └── template.py │ │ └── Host │ │ │ ├── AliasLike │ │ │ ├── 00_Input.mlir │ │ │ ├── 01_HostOpt.mlir │ │ │ ├── 02a_ByreHost.mlir │ │ │ ├── 02b_ToLLVM.mlir │ │ │ ├── 03b_ToLLVMIR.mlir │ │ │ ├── Output.ll │ │ │ ├── Output.mlir │ │ │ ├── TotalPipeline.mlir │ │ │ └── template.py │ │ │ ├── Case0 │ │ │ ├── 00_Input.mlir │ │ │ ├── 01_HostOpt.mlir │ │ │ ├── 02a_ByreHost.mlir │ │ │ ├── 02b_ToLLVM.mlir │ │ │ ├── 03b_ToLLVMIR.mlir │ │ │ ├── Output.ll │ │ │ ├── Output.mlir │ │ │ ├── TotalPipeline.mlir │ │ │ └── template.py │ │ │ ├── Case0_Bytecode │ │ │ ├── 00_Input.mlir │ │ │ ├── 01_HostOpt.mlir │ │ │ ├── 02a_ByreHost.mlir │ │ │ ├── 02b_ToLLVM.mlir │ │ │ ├── 03a_ByreSerial.mlir │ │ │ ├── 03b_ToLLVMBC.mlir │ │ │ ├── Output.bc │ │ │ ├── Output.mlirbc │ │ │ └── template.py │ │ │ ├── Case1 │ │ │ ├── 00_Input.mlir │ │ │ ├── 01_HostOpt.mlir │ │ │ ├── 02a_ByreHost.mlir │ │ │ ├── 02b_ToLLVM.mlir │ │ │ ├── 03b_ToLLVMIR.mlir │ │ │ ├── Output.ll │ │ │ ├── Output.mlir │ │ │ ├── TotalPipeline.mlir │ │ │ └── template.py │ │ │ ├── RngNormal │ │ │ ├── 00_Input.mlir │ │ │ ├── 01_HostOpt.mlir │ │ │ ├── 02a_ByreHost.mlir │ │ │ ├── 02b_ToLLVM.mlir │ │ │ ├── 03b_ToLLVMIR.mlir │ │ │ ├── Output.ll │ │ │ ├── Output.mlir │ │ │ ├── TotalPipeline.mlir │ │ │ └── template.py │ │ │ ├── RngUniform │ │ │ ├── 00_Input.mlir │ │ │ ├── 01_HostOpt.mlir │ │ │ ├── 02a_ByreHost.mlir │ │ │ ├── 02b_ToLLVM.mlir │ │ │ ├── 03b_ToLLVMIR.mlir │ │ │ ├── Output.ll │ │ │ ├── Output.mlir │ │ │ ├── TotalPipeline.mlir │ │ │ └── template.py │ │ │ ├── Transpose │ │ │ ├── 00_Input.mlir │ │ │ ├── 01_HostOpt.mlir │ │ │ ├── 02a_ByreHost.mlir │ │ │ ├── 02b_ToLLVM.mlir │ │ │ ├── 03b_ToLLVMIR.mlir │ │ │ ├── Output.ll │ │ │ ├── Output.mlir │ │ │ ├── TotalPipeline.mlir │ │ │ └── template.py │ │ │ └── TypeCvt │ │ │ ├── 00_Input.mlir │ │ │ ├── 01_HostOpt.mlir │ │ │ ├── 02a_ByreHost.mlir │ │ │ ├── 02b_ToLLVM.mlir │ │ │ ├── 03b_ToLLVMIR.mlir │ │ │ ├── Output.ll │ │ │ ├── Output.mlir │ │ │ ├── TotalPipeline.mlir │ │ │ └── template.py │ ├── Ops │ │ ├── conv.mlir │ │ └── dot.mlir │ ├── Pipelines │ │ ├── BufferizeOpts │ │ │ ├── linalg-ext.mlir │ │ │ └── tensor.mlir │ │ ├── HloOpts │ │ │ ├── mlp.mlir │ │ │ └── rng.mlir │ │ ├── Host │ │ │ ├── Codegen │ │ │ │ └── transpose.mlir │ │ │ └── ToLLVM │ │ │ │ ├── subview.mlir │ │ │ │ └── tanh.mlir │ │ ├── LinalgTensorOpt │ │ │ ├── elementwiseCodegen.mlir │ │ │ └── reductionCodegen.mlir │ │ └── ShapeOpts │ │ │ └── dynamicPartitionStitch.mlir │ ├── Stat │ │ ├── allocCnt.mlir │ │ ├── opCnt.mlir │ │ └── opTypes.mlir │ ├── Target │ │ ├── CUDA │ │ │ ├── all.mlir │ │ │ └── kernel.mlir │ │ ├── Cpp │ │ │ ├── attrs.mlir │ │ │ ├── binary.mlir │ │ │ ├── call.mlir │ │ │ ├── cast.mlir │ │ │ ├── common-cpp.mlir │ │ │ ├── const.mlir │ │ │ ├── control_flow.mlir │ │ │ ├── for.mlir │ │ │ ├── if.mlir │ │ │ ├── invalid.mlir │ │ │ ├── memref.mlir │ │ │ ├── opaque_types.mlir │ │ │ ├── stdops.mlir │ │ │ └── types.mlir │ │ └── PTX │ │ │ └── fusionFuncToPTX.mlir │ ├── Transforms │ │ ├── ApplyPDLPatterns │ │ │ ├── Case_0.mlir │ │ │ └── Pattern_0.mlir │ │ ├── CanonicalizeExt │ │ │ ├── basic.mlir │ │ │ ├── broadcast.mlir │ │ │ ├── concat.mlir │ │ │ ├── deprecated.mlir │ │ │ ├── elementwise.mlir │ │ │ ├── gather.mlir │ │ │ ├── reduce_like.mlir │ │ │ ├── slice_concat.mlir │ │ │ └── transpose.mlir │ │ ├── boundedShapeInference.mlir │ │ ├── cmae.mlir │ │ ├── collectFunc.mlir │ │ ├── funTag.mlir │ │ ├── genericDeviceConfig.mlir │ │ ├── gereicDeviceConfig_with_ByreOpt.mlir │ │ ├── graphCanonicalize.mlir │ │ ├── graphClusteringByDevice.mlir │ │ ├── graphClusteringByDeviceBottomUp.mlir │ │ ├── graphClusteringByDeviceGreedy.mlir │ │ ├── graphClusteringByDeviceTopDown.mlir │ │ ├── insertUniqueId.mlir │ │ ├── insertUniqueIdErase.mlir │ │ ├── loopTag.mlir │ │ ├── loopUnrollImperfect.mlir │ │ ├── loopUnrollUseAnchor.mlir │ │ ├── loopUnrollUseDepth.mlir │ │ ├── loopUnrollWithAnnotation.mlir │ │ ├── memoryPlanning.mlir │ │ ├── oneShotBufferize.mlir │ │ ├── oneShotBufferizeOutParams.mlir │ │ ├── removeFunTag.mlir │ │ ├── removeFuncBody.mlir │ │ ├── rewriteOpToStdCall.mlir │ │ ├── setAllSpace.mlir │ │ ├── setArgShape.mlir │ │ ├── setArgSpace.mlir │ │ ├── setArgSpaceAutoDeduce.mlir │ │ ├── setOpAndArgSpace.mlir │ │ ├── setOpSpace.mlir │ │ ├── shapeFuncOutlining.mlir │ │ ├── shapeReification.mlir │ │ ├── staticShapeInference.mlir │ │ └── testGraphClusteringByDeviceOpNum.mlir │ ├── Utils │ │ ├── testMergeTwoModulesCase0.mlir │ │ ├── testMergeTwoModulesCase0_1.mlir │ │ ├── testMergeTwoModulesCase1.mlir │ │ ├── testMergeTwoModulesCase1_1.mlir │ │ ├── testMergeTwoModulesCase2.mlir │ │ └── testMergeTwoModulesCase2_1.mlir │ ├── lib │ │ ├── Analysis │ │ │ ├── CMakeLists.txt │ │ │ ├── TestGraphClusteringByDeviceOpNum.cpp │ │ │ ├── TestPrintLiveness.cpp │ │ │ ├── TestPrintShapeAnalysis.cpp │ │ │ ├── TestPrintSideEffect.cpp │ │ │ ├── TestPrintSymbolicShape.cpp │ │ │ └── TestPrintUseRange.cpp │ │ ├── CMakeLists.txt │ │ ├── Interface │ │ │ ├── CMakeLists.txt │ │ │ └── TestByreOpInterface.cpp │ │ ├── Transformation │ │ │ ├── CMakeLists.txt │ │ │ ├── TestByreSerialRoundtrip.cpp │ │ │ ├── TestConvertFuncToCustomCall.cpp │ │ │ ├── TestConvertInsertion.cpp │ │ │ ├── TestDTypeConversion.cpp │ │ │ └── TestFuncArgRearrangement.cpp │ │ └── Utils │ │ │ ├── CMakeLists.txt │ │ │ ├── TestBroadcastDenseElementsAttr.cpp │ │ │ └── TestMergeTwoModules.cpp │ ├── lit.cfg.py │ └── lit.site.cfg.py.in └── tools │ ├── CMakeLists.txt │ ├── byteir-cpu-runner │ ├── CMakeLists.txt │ └── byteir-cpu-runner.cpp │ ├── byteir-opt │ ├── CMakeLists.txt │ └── byteir-opt.cpp │ ├── byteir-stat │ ├── CMakeLists.txt │ └── byteir-stat.cpp │ └── byteir-translate │ ├── CMakeLists.txt │ └── byteir-translate.cpp ├── docker └── Dockerfile ├── external ├── TritonTemplate │ ├── .gitignore │ ├── README.md │ └── python │ │ ├── setup.py │ │ └── tritontemplate │ │ ├── __init__.py │ │ ├── _libinfo.py │ │ ├── backend │ │ ├── __init__.py │ │ └── cuda │ │ │ ├── __init__.py │ │ │ ├── bmm │ │ │ ├── __init__.py │ │ │ └── bmm.py │ │ │ ├── gemm │ │ │ ├── __init__.py │ │ │ └── gemm.py │ │ │ ├── layernorm │ │ │ ├── __init__.py │ │ │ └── layernorm.py │ │ │ ├── softmax │ │ │ ├── __init__.py │ │ │ └── softmax.py │ │ │ ├── transpose │ │ │ ├── __init__.py │ │ │ ├── transpose_0213.py │ │ │ └── transpose_10.py │ │ │ └── utils │ │ │ ├── __init__.py │ │ │ ├── activation.py │ │ │ └── utils.py │ │ ├── compiler │ │ ├── __init__.py │ │ ├── base.py │ │ ├── compiler.py │ │ ├── dtype.py │ │ ├── kernel.py │ │ ├── op_registry.py │ │ ├── ops │ │ │ ├── __init__.py │ │ │ ├── bmm │ │ │ │ ├── __init__.py │ │ │ │ └── bmm.py │ │ │ ├── gemm │ │ │ │ ├── __init__.py │ │ │ │ └── gemm.py │ │ │ ├── layernorm │ │ │ │ ├── __init__.py │ │ │ │ └── layernorm.py │ │ │ ├── softmax │ │ │ │ ├── __init__.py │ │ │ │ └── softmax.py │ │ │ └── transpose │ │ │ │ ├── __init__.py │ │ │ │ └── transpose.py │ │ └── utils.py │ │ ├── testing │ │ ├── __init__.py │ │ ├── aot_demo.py │ │ ├── cuda │ │ │ ├── test_bmm.py │ │ │ ├── test_gemm.py │ │ │ ├── test_layernorm.py │ │ │ ├── test_softmax.py │ │ │ └── test_transpose.py │ │ └── ptx_gen_demo.py │ │ └── utils │ │ ├── __init__.py │ │ ├── tensor_utils.py │ │ └── torch_utils.py ├── half │ ├── LICENSE.txt │ ├── README.txt │ └── include │ │ └── half │ │ └── half.hpp └── patches │ └── AITemplate │ ├── A10.patch │ ├── logging.patch │ └── num_builders.patch ├── external_libs └── runtime │ ├── CMakeLists.txt │ ├── README.md │ └── flash_attn │ ├── CMakeLists.txt │ ├── include │ └── flash_api.h │ └── lib │ ├── CMakeLists.txt │ ├── alibi.h │ ├── block_info.h │ ├── dropout.h │ ├── flash.h │ ├── flash_api.cu │ ├── flash_bwd_hdim128_fp16_sm80.cu │ ├── flash_bwd_hdim160_fp16_sm80.cu │ ├── flash_bwd_hdim192_fp16_sm80.cu │ ├── flash_bwd_hdim224_fp16_sm80.cu │ ├── flash_bwd_hdim256_fp16_sm80.cu │ ├── flash_bwd_hdim32_fp16_sm80.cu │ ├── flash_bwd_hdim64_fp16_sm80.cu │ ├── flash_bwd_hdim96_fp16_sm80.cu │ ├── flash_bwd_kernel.h │ ├── flash_bwd_launch_template.h │ ├── flash_bwd_preprocess_kernel.h │ ├── flash_fwd_hdim128_fp16_sm80.cu │ ├── flash_fwd_hdim160_fp16_sm80.cu │ ├── flash_fwd_hdim192_fp16_sm80.cu │ ├── flash_fwd_hdim224_fp16_sm80.cu │ ├── flash_fwd_hdim256_fp16_sm80.cu │ ├── flash_fwd_hdim32_fp16_sm80.cu │ ├── flash_fwd_hdim64_fp16_sm80.cu │ ├── flash_fwd_hdim96_fp16_sm80.cu │ ├── flash_fwd_kernel.h │ ├── flash_fwd_launch_template.h │ ├── flash_fwd_split_hdim128_fp16_sm80.cu │ ├── flash_fwd_split_hdim160_fp16_sm80.cu │ ├── flash_fwd_split_hdim192_fp16_sm80.cu │ ├── flash_fwd_split_hdim224_fp16_sm80.cu │ ├── flash_fwd_split_hdim256_fp16_sm80.cu │ ├── flash_fwd_split_hdim32_fp16_sm80.cu │ ├── flash_fwd_split_hdim64_fp16_sm80.cu │ ├── flash_fwd_split_hdim96_fp16_sm80.cu │ ├── kernel_traits.h │ ├── mask.h │ ├── philox.cuh │ ├── rotary.h │ ├── softmax.h │ ├── static_switch.h │ └── utils.h ├── frontends ├── README.md ├── onnx-frontend │ ├── .gitignore │ ├── CMakeLists.txt │ ├── MLIR.cmake │ ├── README.md │ ├── onnx-frontend │ │ ├── CMakeLists.txt │ │ ├── src │ │ │ ├── CMakeLists.txt │ │ │ ├── Compiler │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── OFCompilerOptions.cpp │ │ │ │ ├── OFCompilerOptions.hpp │ │ │ │ ├── OFCompilerPipelines.cpp │ │ │ │ ├── OFCompilerPipelines.hpp │ │ │ │ ├── OFCompilerTypes.hpp │ │ │ │ ├── OFCompilerUtils.cpp │ │ │ │ └── OFCompilerUtils.hpp │ │ │ ├── Conversion │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── OFCanonicalizer.cpp │ │ │ │ ├── OFCanonicalizer.hpp │ │ │ │ ├── OFCheckNonLowered.cpp │ │ │ │ ├── OFCheckNonLowered.hpp │ │ │ │ ├── OFInsertNecessaryCast.cpp │ │ │ │ ├── OFInsertNecessaryCast.hpp │ │ │ │ ├── OFModifyEntryPoint.cpp │ │ │ │ ├── OFModifyEntryPoint.hpp │ │ │ │ ├── OFPasses.hpp │ │ │ │ ├── OFPasses.td │ │ │ │ ├── OFPassesDetail.hpp │ │ │ │ ├── OFRewriteCustomOnnxOps.cpp │ │ │ │ ├── OFRewriteCustomOnnxOps.hpp │ │ │ │ ├── OFRewriteCustomOnnxOps.td │ │ │ │ ├── OFRewriteToCustomCall.cpp │ │ │ │ ├── OFRewriteToCustomCall.hpp │ │ │ │ └── OFRewriteToCustomCall.td │ │ │ ├── Support │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── OFConstants.hpp │ │ │ │ ├── OFUtils.cpp │ │ │ │ └── OFUtils.hpp │ │ │ ├── onnx-frontend-opt.cpp │ │ │ └── onnx-frontend.cpp │ │ └── test │ │ │ ├── CMakeLists.txt │ │ │ ├── dynamic_shape_relu.onnx │ │ │ ├── lit.cfg.py │ │ │ ├── lit.site.cfg.py.in │ │ │ ├── of_canonicalizer.mlir │ │ │ ├── of_check_non_lowered.mlir │ │ │ ├── of_modify_entry_point.mlir │ │ │ ├── of_rewrite_custom_onnx_op.mlir │ │ │ ├── of_rewrite_to_custom_call.mlir │ │ │ ├── set_shape.mlir │ │ │ └── shape_inference.mlir │ ├── pytest.ini │ ├── requirements.txt │ ├── scripts │ │ ├── build_and_test.sh │ │ └── envsetup.sh │ ├── test │ │ ├── __init__.py │ │ ├── base.py │ │ ├── env.py │ │ ├── models │ │ │ ├── test_batch_size.py │ │ │ └── test_large_model.py │ │ └── ops │ │ │ ├── __init__.py │ │ │ ├── data │ │ │ ├── math │ │ │ │ ├── clip.onnx │ │ │ │ ├── gelu.onnx │ │ │ │ └── softmax.onnx │ │ │ ├── nn │ │ │ │ └── batch_normalization.onnx │ │ │ ├── quantize │ │ │ │ └── quantize_dequantize.onnx │ │ │ └── tensor │ │ │ │ ├── arg_max.onnx │ │ │ │ ├── arg_min.onnx │ │ │ │ ├── concat.onnx │ │ │ │ ├── concat_dynamic_shape.onnx │ │ │ │ └── resize_nearest_v10.onnx │ │ │ ├── test_math.py │ │ │ ├── test_nn.py │ │ │ ├── test_quantize.py │ │ │ ├── test_rnn.py │ │ │ ├── test_tensor.py │ │ │ └── utils.py │ └── third_party │ │ └── patches │ │ ├── OnnxMlirConcat.patch │ │ ├── OnnxMlirConvTranspose.patch │ │ ├── OnnxMlirDialectBuilder.patch │ │ ├── OnnxMlirDialectRewrite.patch │ │ ├── OnnxMlirDynamicShape.patch │ │ ├── OnnxMlirElementwise.patch │ │ ├── OnnxMlirKeepCustomOpType.patch │ │ ├── OnnxMlirMatMul.patch │ │ ├── OnnxMlirONNXToStablehloCommon.patch │ │ ├── OnnxMlirONNXToStablehloGather.patch │ │ ├── OnnxMlirOnnxOpsTensorPad.patch │ │ ├── OnnxMlirPooling.patch │ │ ├── OnnxMlirReductionUpgrade.patch │ │ ├── OnnxMlirRegisterLibrary.patch │ │ ├── OnnxMlirReshape.patch │ │ ├── OnnxMlirScatterElements.patch │ │ ├── OnnxMlirTestElementwise.patch │ │ ├── OnnxMlirTestPooling.patch │ │ ├── OnnxMlirWillNotPush.patch │ │ ├── OnnxMlirWillPushToUpstream.patch │ │ ├── OnnxOfficialExternalData.patch │ │ └── OnnxOfficialResize.patch ├── tf-frontend │ ├── .bazelrc │ ├── .bazelversion │ ├── .gitignore │ ├── .tf_configure.bazelrc │ ├── BUILD │ ├── README.md │ ├── WORKSPACE │ ├── byteir │ │ ├── BUILD │ │ ├── ace.BUILD │ │ └── workspace.bzl │ ├── docs │ │ ├── attributes.md │ │ └── developer_guild.md │ ├── example │ │ ├── resnet.py │ │ └── resnet50_model.py │ ├── external │ │ └── patches │ │ │ └── tensorflow │ │ │ ├── fix-bug-of-create-f16-const-for-HoistCwiseBinaryOutO.patch │ │ │ ├── for_gcc_8_5.patch │ │ │ ├── grappler.patch │ │ │ ├── mhlo_ops.patch │ │ │ ├── support-tf-shape-inference.patch │ │ │ ├── tf.Select_to_mhlo.select.patch │ │ │ ├── tf_build.patch │ │ │ ├── tf_dilated_conv.patch │ │ │ ├── tf_mkl.patch │ │ │ ├── tf_slice.patch │ │ │ └── topk.patch │ ├── scripts │ │ ├── apply_patches.sh │ │ ├── build_and_test.sh │ │ └── prepare.sh │ ├── tf_mlir_ext │ │ ├── numerical │ │ │ ├── BUILD │ │ │ ├── dilated_conv2d.mlir │ │ │ ├── fallback_to_custom_call.mlir │ │ │ ├── fuse_tf_ops.mlir │ │ │ ├── glob_lit_test.bzl │ │ │ ├── numerical_test.py │ │ │ ├── process_dynamic_stitch_as_static.mlir │ │ │ ├── reshape_movedown_string.mlir │ │ │ ├── rewrite_to_custom_call.mlir │ │ │ ├── runlit.cfg.py │ │ │ ├── runlit.site.cfg.py │ │ │ └── where.mlir │ │ ├── pipelines │ │ │ ├── BUILD │ │ │ ├── customized_tf_to_mhlo.cc │ │ │ ├── customized_tf_to_mhlo.h │ │ │ ├── passes.h │ │ │ ├── passes.td │ │ │ └── passes_detail.h │ │ ├── tests │ │ │ ├── BUILD │ │ │ ├── ace_ops.mlir │ │ │ ├── convert_repeat_to_tile.mlir │ │ │ ├── dilated_conv2d.mlir │ │ │ ├── fallback_to_custom_call.mlir │ │ │ ├── fuse_tf_ops.mlir │ │ │ ├── glob_lit_test.bzl │ │ │ ├── inline_func_call_in_scf_if.mlir │ │ │ ├── mhlo_legalize_tf_ext.mlir │ │ │ ├── process_dynamic_stitch_as_static.mlir │ │ │ ├── reshape_movedown_string.mlir │ │ │ ├── rewrite_func_attr_to_byteir.mlir │ │ │ ├── rewrite_to_custom_call.mlir │ │ │ ├── rewrite_to_custom_call_keep_body.mlir │ │ │ ├── rewrite_to_if.mlir │ │ │ ├── runlit.cfg.py │ │ │ ├── runlit.site.cfg.py │ │ │ ├── set_repeat_out_batch_size.mlir │ │ │ └── where.mlir │ │ ├── transforms │ │ │ ├── BUILD │ │ │ ├── constant_folding.cc │ │ │ ├── constant_folding.h │ │ │ ├── convert_repeat_to_tile.cc │ │ │ ├── convert_repeat_to_tile.h │ │ │ ├── fuse_tf_ops.cc │ │ │ ├── fuse_tf_ops.h │ │ │ ├── fuse_tf_ops.td │ │ │ ├── inline_func_call_in_scf_if.cc │ │ │ ├── inline_func_call_in_scf_if.h │ │ │ ├── mhlo_legalize_tf_ext.cc │ │ │ ├── mhlo_legalize_tf_ext.h │ │ │ ├── passes.h │ │ │ ├── passes.td │ │ │ ├── passes_detail.h │ │ │ ├── process_dynamic_stitch_as_static.cc │ │ │ ├── process_dynamic_stitch_as_static.h │ │ │ ├── remove_control_flow.cc │ │ │ ├── remove_control_flow.h │ │ │ ├── reshape_movedown_string.cc │ │ │ ├── reshape_movedown_string.h │ │ │ ├── rewrite_func_attr_to_byteir.cc │ │ │ ├── rewrite_func_attr_to_byteir.h │ │ │ ├── rewrite_to_custom_call.cc │ │ │ ├── rewrite_to_custom_call.h │ │ │ ├── rewrite_to_custom_call.td │ │ │ ├── rewrite_to_if.cc │ │ │ ├── rewrite_to_if.h │ │ │ ├── set_repeat_out_batch_size.cc │ │ │ ├── set_repeat_out_batch_size.h │ │ │ ├── tf_fallback_to_custom_call.cc │ │ │ ├── tf_fallback_to_custom_call.h │ │ │ ├── tf_switch_merge_to_if.cc │ │ │ └── tf_switch_merge_to_if.h │ │ └── utils │ │ │ ├── BUILD │ │ │ ├── customcall.cc │ │ │ ├── customcall.h │ │ │ ├── dce.cc │ │ │ ├── dce.h │ │ │ ├── utils.cc │ │ │ └── utils.h │ ├── tools │ │ ├── BUILD │ │ ├── tf_ext_opt_main.cc │ │ └── tf_frontend_main.cc │ └── utils │ │ ├── BUILD │ │ ├── attributes.h │ │ ├── graphdef_opt.cc │ │ ├── graphdef_opt.h │ │ ├── misc.cc │ │ └── misc.h └── torch-frontend │ ├── .gitignore │ ├── CMakeLists.txt │ ├── MLIR.cmake │ ├── README.md │ ├── TorchMLIR.cmake │ ├── build-requirements.txt │ ├── doc │ └── torch_2_0_training.md │ ├── examples │ ├── demo │ │ ├── README.md │ │ ├── backend.py │ │ ├── byteir_fusible_pattern.py │ │ ├── compile_utils.py │ │ ├── config.py │ │ ├── context.py │ │ ├── fx_match_utils.py │ │ ├── main.py │ │ ├── models │ │ │ ├── configuration_chatglm.py │ │ │ ├── modeling_chatglm.py │ │ │ └── modeling_nanogpt.py │ │ ├── partitioners.py │ │ └── requirements.txt │ ├── inference │ │ ├── brt_backend.py │ │ ├── infer_resnet.py │ │ ├── infer_tinybert.py │ │ ├── mixtral │ │ │ ├── infer_single_mixtral.py │ │ │ └── requirements.txt │ │ ├── mlp.py │ │ └── tit_mlp.py │ └── training │ │ ├── byteir_backend.py │ │ ├── mlp.py │ │ ├── train_resnet.py │ │ └── train_tinybert.py │ ├── scripts │ ├── build.sh │ ├── build_and_test.sh │ └── envsetup.sh │ ├── test-requirements.txt │ ├── third_party │ ├── llvm_patches │ │ └── ir_printing.patch │ └── patches │ │ ├── backend_contract.patch │ │ ├── build.patch │ │ ├── communication_op.patch │ │ ├── custom_op.patch │ │ ├── fx_importer.patch │ │ ├── generated_torch_ops_td.patch │ │ ├── pipeline.patch │ │ ├── reduce_op_variants.patch │ │ └── tuple.patch │ ├── torch-cpu-requirements.txt │ ├── torch-cuda-requirements.txt │ └── torch-frontend │ ├── CMakeLists.txt │ ├── include │ ├── torch-frontend-c │ │ └── Passes.h │ └── torch-frontend │ │ ├── CMakeLists.txt │ │ ├── Conversion │ │ ├── CMakeLists.txt │ │ ├── ConvertTorchToCcl.h │ │ ├── ConvertTorchToCustomCall.h │ │ ├── ConvertTorchToStablehloExt.h │ │ ├── Passes.h │ │ └── Passes.td │ │ ├── Dialect │ │ └── Torch │ │ │ └── Transforms │ │ │ ├── CMakeLists.txt │ │ │ ├── DecomposeOnTorch.h │ │ │ ├── FuseOpOnTorch.h │ │ │ ├── Passes.h │ │ │ └── Passes.td │ │ ├── Pipelines │ │ └── Pipelines.h │ │ ├── Transforms │ │ ├── CMakeLists.txt │ │ ├── CanonicalizeExt.h │ │ ├── EliminateUselessOp.h │ │ ├── Passes.h │ │ ├── Passes.td │ │ ├── RewriteCustomOp.h │ │ ├── RewriteEntryFuncName.h │ │ └── UnpackPublicFunctionReturn.h │ │ └── Utils │ │ ├── ConvertOpFolder.h │ │ └── CustomCallUtil.h │ ├── lib │ ├── CAPI │ │ ├── CMakeLists.txt │ │ └── Passes.cpp │ ├── CMakeLists.txt │ ├── Conversion │ │ ├── CMakeLists.txt │ │ ├── ConvertTorchToCcl.cpp │ │ ├── ConvertTorchToCustomCall.cpp │ │ ├── ConvertTorchToStablehloExt.cpp │ │ └── PassDetail.h │ ├── CustomOp │ │ ├── CMakeLists.txt │ │ ├── dynamic_mask_stitch.cpp │ │ ├── dynamic_partition.cpp │ │ └── dynamic_stitch.cpp │ ├── Dialect │ │ └── Torch │ │ │ └── Transforms │ │ │ ├── CMakeLists.txt │ │ │ ├── DecomposeOnTorch.cpp │ │ │ ├── FuseOpOnTorch.cpp │ │ │ ├── FuseOpOnTorchPattern.td │ │ │ └── PassDetail.h │ ├── Pipelines │ │ ├── CMakeLists.txt │ │ └── Pipelines.cpp │ ├── Transforms │ │ ├── CMakeLists.txt │ │ ├── CanonicalizeExt.cpp │ │ ├── EliminateUselessOp.cpp │ │ ├── PassDetail.h │ │ ├── RewriteCustomOp.cpp │ │ ├── RewriteEntryFuncName.cpp │ │ └── UnpackPublicFunctionReturn.cpp │ └── Utils │ │ ├── CMakeLists.txt │ │ └── ConvertOpFolder.cpp │ ├── python │ ├── CMakeLists.txt │ ├── TorchFrontendModule.cpp │ ├── setup.py │ ├── test │ │ ├── pytest.ini │ │ ├── test_attn_rewrite.py │ │ ├── test_fx_utils.py │ │ ├── test_fximporter │ │ │ ├── test_ccl.py │ │ │ ├── test_custom_ops.py │ │ │ ├── test_ops_fximporter.py │ │ │ └── utils.py │ │ ├── test_math_custom_ops.py │ │ ├── test_stablehlo_bytecode.py │ │ ├── test_torchscript │ │ │ ├── test_byteir_customcall_ops.py │ │ │ ├── test_compile_option.py │ │ │ ├── test_model.py │ │ │ ├── test_ops.py │ │ │ └── test_torch_custom_ops.py │ │ └── test_utils │ │ │ └── test_jit_transforms.py │ ├── torch_frontend │ │ ├── __init__.py │ │ ├── _mlir_libs │ │ │ └── _site_initialize_0.py │ │ ├── byteir_backend │ │ │ ├── README.md │ │ │ ├── __init__.py │ │ │ ├── byteir_fusible_pattern.py │ │ │ ├── compilation_cache.py │ │ │ ├── compiled_function.py │ │ │ ├── compiler.py │ │ │ ├── config.py │ │ │ ├── debug.py │ │ │ ├── fx_match_utils.py │ │ │ ├── fx_utils.py │ │ │ ├── inner_compile.py │ │ │ ├── partitioners.py │ │ │ └── utils.py │ │ ├── compile.py │ │ ├── extra_shape_fn.py │ │ ├── flash_attn_op.py │ │ ├── fx_rewrite.py │ │ ├── fx_tracer.py │ │ ├── fx_utils.py │ │ ├── tools │ │ │ ├── compiler.py │ │ │ ├── extra_fn.mlir │ │ │ └── gen_extra_library.py │ │ ├── ts_utils.py │ │ └── utils │ │ │ ├── __init__.py │ │ │ └── jit_transforms.py │ └── version.txt │ ├── test │ ├── CMakeLists.txt │ ├── Conversion │ │ ├── ConvertTorchToCcl.mlir │ │ ├── ConvertTorchToCustomCall.mlir │ │ └── ConvertTorchToStablehloExt.mlir │ ├── Dialect │ │ └── Torch │ │ │ ├── DecomposeOnTorch.mlir │ │ │ └── FuseOpOnTorch.mlir │ ├── Pipelines │ │ └── TorchFunctionToTorchPipeline.mlir │ ├── Transforms │ │ ├── EliminateUselessOp.mlir │ │ ├── RewriteEntryFuncName.mlir │ │ └── UnpackPublicFunctionReturn.mlir │ ├── lit.cfg.py │ └── lit.site.cfg.py.in │ └── tools │ ├── CMakeLists.txt │ └── torch-frontend-opt.cpp ├── runtime ├── .gitignore ├── README.md ├── VERSION_NUMBER ├── cmake │ ├── CMakeLists.txt │ ├── Modules │ │ └── FindNCCL.cmake │ ├── brt_common.cmake │ ├── brt_config.h.in │ ├── brt_device_cpu.cmake │ ├── brt_device_cuda.cmake │ ├── brt_device_nccl.cmake │ ├── brt_framework.cmake │ ├── brt_ir.cmake │ ├── brt_provider_cpu.cmake │ ├── brt_provider_cuda.cmake │ ├── brt_provider_nccl.cmake │ ├── brt_python_bindings.cmake │ ├── brt_shared.cmake │ └── brt_unittests.cmake ├── examples │ └── external_project │ │ ├── CMakeLists.txt │ │ └── main.cpp ├── include │ └── brt │ │ ├── backends │ │ ├── README.md │ │ ├── common.h │ │ ├── cpu │ │ │ ├── device │ │ │ │ ├── cpu_device_api.h │ │ │ │ ├── cpu_work_queue.h │ │ │ │ └── llvm │ │ │ │ │ └── jit.h │ │ │ └── providers │ │ │ │ └── default │ │ │ │ └── cpu_provider.h │ │ ├── cuda │ │ │ ├── device │ │ │ │ ├── common │ │ │ │ │ ├── cuda_call.h │ │ │ │ │ ├── dtype.h │ │ │ │ │ ├── fast_divmod.h │ │ │ │ │ └── util.h │ │ │ │ ├── compile │ │ │ │ │ ├── nvrtc.h │ │ │ │ │ └── ptx.h │ │ │ │ ├── cuda_allocator.h │ │ │ │ ├── cuda_device_api.h │ │ │ │ ├── cuda_env.h │ │ │ │ ├── cuda_work_queue.h │ │ │ │ └── utils │ │ │ │ │ └── op_kernel_impl_helpers.h │ │ │ └── providers │ │ │ │ └── default │ │ │ │ ├── ait │ │ │ │ ├── model_interface.h │ │ │ │ └── op_registration.h │ │ │ │ ├── codegen │ │ │ │ └── op_registration.h │ │ │ │ ├── copy │ │ │ │ └── op_registration.h │ │ │ │ ├── cuda_provider.h │ │ │ │ ├── cudnn_helper.h │ │ │ │ ├── custom │ │ │ │ └── op_registration.h │ │ │ │ ├── indexing │ │ │ │ └── op_registration.h │ │ │ │ ├── math │ │ │ │ ├── helper.h │ │ │ │ └── op_registration.h │ │ │ │ ├── normalization │ │ │ │ └── op_registration.h │ │ │ │ ├── reduction │ │ │ │ └── op_registration.h │ │ │ │ ├── tensor_generate │ │ │ │ └── op_registration.h │ │ │ │ └── tensor_manipulate │ │ │ │ └── op_registration.h │ │ ├── nccl │ │ │ ├── device │ │ │ │ ├── d_context_nccl.h │ │ │ │ ├── distributed_backend_nccl.h │ │ │ │ └── utils.h │ │ │ └── providers │ │ │ │ ├── nccl_provider.h │ │ │ │ └── op_registration.h │ │ └── rng_state_context.h │ │ └── core │ │ ├── common │ │ ├── code_location.h │ │ ├── common.h │ │ ├── enums.h │ │ ├── exceptions.h │ │ ├── logging │ │ │ ├── capture.h │ │ │ ├── isink.h │ │ │ ├── logging.h │ │ │ ├── macros.h │ │ │ ├── severity.h │ │ │ └── sinks │ │ │ │ ├── cerr_sink.h │ │ │ │ ├── clog_sink.h │ │ │ │ ├── composite_sink.h │ │ │ │ ├── file_sink.h │ │ │ │ └── ostream_sink.h │ │ ├── make_string.h │ │ ├── status.h │ │ ├── string_view.h │ │ └── utils │ │ │ └── math_helper.h │ │ ├── context │ │ ├── execution_context.h │ │ ├── execution_frame.h │ │ └── work_queue.h │ │ ├── distributed │ │ ├── d_context.h │ │ ├── distributed_backend.h │ │ ├── distributed_session.h │ │ └── rendezvous_socket.h │ │ ├── framework │ │ ├── allocator.h │ │ ├── arena.h │ │ ├── bfc_arena.h │ │ ├── brt_mutex.h │ │ ├── device_api.h │ │ ├── dtype.h │ │ ├── event.h │ │ ├── execution_plan.h │ │ ├── execution_provider.h │ │ ├── kernel_registry.h │ │ ├── memory_info.h │ │ ├── op_accessor.h │ │ ├── op_kernel.h │ │ ├── op_kernel_impl_base.h │ │ ├── op_kernel_info.h │ │ └── value.h │ │ ├── ir │ │ ├── builder.h │ │ ├── engine_util.h │ │ ├── graph_info.h │ │ ├── ir.h │ │ ├── op_helper.h │ │ └── util.h │ │ └── session │ │ ├── request_context.h │ │ └── session.h ├── lib │ ├── backends │ │ ├── cpu │ │ │ ├── device │ │ │ │ ├── cpu_device_api.cc │ │ │ │ ├── cpu_work_queue.cc │ │ │ │ └── llvm │ │ │ │ │ └── jit.cc │ │ │ └── providers │ │ │ │ └── default │ │ │ │ ├── copy │ │ │ │ ├── copy.cc │ │ │ │ └── copy.h │ │ │ │ ├── cpu_provider.cc │ │ │ │ ├── custom_call │ │ │ │ ├── non_zero.cc │ │ │ │ ├── non_zero.h │ │ │ │ ├── repeat.cc │ │ │ │ ├── repeat.h │ │ │ │ ├── tf_equal.cc │ │ │ │ ├── tf_equal.h │ │ │ │ ├── tf_select.cc │ │ │ │ ├── tf_select.h │ │ │ │ ├── tf_string_to_number.cc │ │ │ │ ├── tf_string_to_number.h │ │ │ │ ├── topk.cc │ │ │ │ └── topk.h │ │ │ │ ├── llvm │ │ │ │ ├── jit.cc │ │ │ │ └── jit.h │ │ │ │ ├── math │ │ │ │ ├── elementwise_ops.cc │ │ │ │ └── elementwise_ops.h │ │ │ │ ├── shape │ │ │ │ ├── shape_compute.cc │ │ │ │ └── shape_compute.h │ │ │ │ ├── tensor_generate │ │ │ │ ├── fill.cc │ │ │ │ ├── fill.h │ │ │ │ ├── rng_state.cc │ │ │ │ └── rng_state.h │ │ │ │ └── typecvt │ │ │ │ └── typecvt.h │ │ ├── cuda │ │ │ ├── device │ │ │ │ ├── common │ │ │ │ │ ├── cuda_call.cc │ │ │ │ │ └── util.cc │ │ │ │ ├── compile │ │ │ │ │ ├── nvrtc.cc │ │ │ │ │ └── ptx.cc │ │ │ │ ├── cuda_allocator.cc │ │ │ │ ├── cuda_device_api.cc │ │ │ │ ├── cuda_env.cc │ │ │ │ └── cuda_work_queue.cc │ │ │ └── providers │ │ │ │ └── default │ │ │ │ ├── ait │ │ │ │ ├── ait.cc │ │ │ │ ├── ait.h │ │ │ │ └── op_registration.cc │ │ │ │ ├── codegen │ │ │ │ ├── op_registration.cc │ │ │ │ ├── ptx.cc │ │ │ │ └── ptx.h │ │ │ │ ├── copy │ │ │ │ ├── copy.cc │ │ │ │ ├── copy.h │ │ │ │ └── op_registration.cc │ │ │ │ ├── cuda_provider.cc │ │ │ │ ├── custom │ │ │ │ ├── custom.cc │ │ │ │ ├── custom.h │ │ │ │ └── op_registration.cc │ │ │ │ ├── indexing │ │ │ │ ├── index_put.h │ │ │ │ ├── index_select.h │ │ │ │ ├── kernels │ │ │ │ │ ├── index_put.cu │ │ │ │ │ ├── index_put.h │ │ │ │ │ ├── index_select.cu │ │ │ │ │ └── index_select.h │ │ │ │ └── op_registration.cc │ │ │ │ ├── math │ │ │ │ ├── batch_matmul.cc │ │ │ │ ├── batch_matmul.h │ │ │ │ ├── conv.cc │ │ │ │ ├── conv.h │ │ │ │ ├── conv_backward.cc │ │ │ │ ├── conv_backward.h │ │ │ │ ├── elementwise_ops.cc │ │ │ │ ├── elementwise_ops.h │ │ │ │ ├── helper.cc │ │ │ │ ├── kernels │ │ │ │ │ ├── cutlass_blas.cu │ │ │ │ │ ├── cutlass_blas.h │ │ │ │ │ ├── elementwise.cu │ │ │ │ │ └── elementwise.h │ │ │ │ ├── matmul.cc │ │ │ │ ├── matmul.h │ │ │ │ ├── op_registration.cc │ │ │ │ ├── pool.cc │ │ │ │ ├── pool.h │ │ │ │ ├── pool_grad.cc │ │ │ │ └── pool_grad.h │ │ │ │ ├── normalization │ │ │ │ ├── batch_norm_grad.cc │ │ │ │ ├── batch_norm_grad.h │ │ │ │ ├── batch_norm_training.cc │ │ │ │ ├── batch_norm_training.h │ │ │ │ └── op_registration.cc │ │ │ │ ├── reduction │ │ │ │ ├── kernels │ │ │ │ │ ├── reduction.cu │ │ │ │ │ ├── reduction.h │ │ │ │ │ └── reduction_helper.h │ │ │ │ ├── op_registration.cc │ │ │ │ └── reduce_impl.h │ │ │ │ ├── tensor_generate │ │ │ │ ├── fill.cc │ │ │ │ ├── fill.h │ │ │ │ ├── kernels │ │ │ │ │ ├── fill.cu │ │ │ │ │ ├── fill.h │ │ │ │ │ ├── rng.cu │ │ │ │ │ └── rng.h │ │ │ │ ├── op_registration.cc │ │ │ │ ├── rng.h │ │ │ │ ├── rng_state.cc │ │ │ │ └── rng_state.h │ │ │ │ └── tensor_manipulate │ │ │ │ ├── kernels │ │ │ │ ├── transpose.cu │ │ │ │ └── transpose.h │ │ │ │ ├── op_registration.cc │ │ │ │ ├── transpose.cc │ │ │ │ └── transpose.h │ │ └── nccl │ │ │ ├── device │ │ │ ├── distributed_backend_nccl.cc │ │ │ └── utils.cc │ │ │ └── providers │ │ │ ├── all_gather.cc │ │ │ ├── all_gather.h │ │ │ ├── all_reduce.cc │ │ │ ├── all_reduce.h │ │ │ ├── broadcast.cc │ │ │ ├── broadcast.h │ │ │ ├── nccl_provider.cc │ │ │ ├── op_registration.cc │ │ │ ├── recv.cc │ │ │ ├── recv.h │ │ │ ├── send.cc │ │ │ └── send.h │ └── core │ │ ├── common │ │ ├── common.cc │ │ ├── logging │ │ │ ├── capture.cc │ │ │ ├── logging.cc │ │ │ └── sinks │ │ │ │ └── ostream_sink.cc │ │ ├── status.cc │ │ └── utils │ │ │ └── math_helper.cc │ │ ├── context │ │ └── execution_frame.cc │ │ ├── distributed │ │ ├── distributed_backend.cc │ │ ├── distributed_session.cc │ │ └── rendezvous_socket.cc │ │ ├── framework │ │ ├── allocator.cc │ │ ├── bfc_arena.cc │ │ ├── device_api.cc │ │ ├── execution_plan.cc │ │ ├── execution_provider.cc │ │ ├── kernel_registry.cc │ │ ├── op_accessor.cc │ │ └── op_kernel_info.cc │ │ ├── ir │ │ ├── builder.cc │ │ ├── ir.cc │ │ ├── op_helper.cc │ │ └── util.cc │ │ └── session │ │ ├── request_context.cc │ │ └── session.cc ├── python │ ├── README.md │ ├── brt │ │ ├── __init__.py │ │ ├── backend.py │ │ └── utils.py │ ├── examples │ │ ├── add2.mlir │ │ ├── add2.py │ │ ├── ait_op.py │ │ ├── arg_alias.mlir │ │ ├── arg_alias.py │ │ ├── distribute_mlp.py │ │ └── llm.py │ ├── setup.py │ └── src │ │ └── module.cc ├── test │ ├── backends │ │ ├── cpu │ │ │ ├── device │ │ │ │ └── llvm_jit_test.cc │ │ │ └── providers │ │ │ │ └── default │ │ │ │ ├── e2e │ │ │ │ └── e2e_test.cc │ │ │ │ ├── kernel │ │ │ │ ├── copy_test.cc │ │ │ │ ├── non_zero_test.cc │ │ │ │ ├── repeat_test.cc │ │ │ │ ├── rng_state_test.cc │ │ │ │ ├── string_equal_test.cc │ │ │ │ ├── tf_select_test.cc │ │ │ │ ├── tf_string_to_number_test.cc │ │ │ │ ├── topk_test.cc │ │ │ │ └── typecvt_test.cc │ │ │ │ └── request_context_test.cc │ │ ├── cuda │ │ │ ├── device │ │ │ │ ├── allocator_test.cc │ │ │ │ ├── cuda_work_queue_test.cc │ │ │ │ ├── nvrtc_test.cc │ │ │ │ ├── ptx_test.cc │ │ │ │ ├── test_kernels.cu │ │ │ │ └── test_kernels.h │ │ │ └── providers │ │ │ │ └── default │ │ │ │ ├── e2e │ │ │ │ └── resnet_test.cc │ │ │ │ ├── kernel │ │ │ │ ├── ait_test.cc │ │ │ │ ├── alias_test.cc │ │ │ │ ├── batch_matmul_test.cc │ │ │ │ ├── batch_norm_grad_test.cc │ │ │ │ ├── batch_norm_training_test.cc │ │ │ │ ├── codegen_test.cc │ │ │ │ ├── conv_backward_data_test.cc │ │ │ │ ├── conv_backward_filter_test.cc │ │ │ │ ├── conv_test.cc │ │ │ │ ├── copy_test.cc │ │ │ │ ├── elementwise_test.cc │ │ │ │ ├── fill_test.cc │ │ │ │ ├── flash_attn_bwd_test.cc │ │ │ │ ├── flash_attn_fwd_test.cc │ │ │ │ ├── index_test.cc │ │ │ │ ├── matmul_test.cc │ │ │ │ ├── multi_stream_test.cc │ │ │ │ ├── pool_grad_test.cc │ │ │ │ ├── pool_test.cc │ │ │ │ ├── reduction_test.cc │ │ │ │ ├── rng_state_test.cc │ │ │ │ ├── rng_test.cc │ │ │ │ └── transpose_test.cc │ │ │ │ ├── request_context_test.cc │ │ │ │ └── session_test.cc │ │ └── nccl │ │ │ ├── device │ │ │ ├── test_distributed_backend.cc │ │ │ └── test_utils.cc │ │ │ └── providers │ │ │ └── test_distributed_session.cc │ ├── common │ │ ├── env.cc │ │ ├── models.cc │ │ └── util.cc │ ├── context │ │ └── exec_frame_test.cc │ ├── distributed │ │ └── test_rendezvous_socket.cc │ ├── exported.ld │ ├── external_kernels │ │ ├── cpu │ │ │ └── kernels.cc │ │ └── cuda │ │ │ ├── kernels.cc │ │ │ ├── kernels.cu │ │ │ └── kernels.h │ ├── framework │ │ ├── allocator_test.cc │ │ └── misc.cc │ ├── include │ │ └── brt │ │ │ └── test │ │ │ └── common │ │ │ ├── config.h │ │ │ ├── cuda │ │ │ └── util.h │ │ │ ├── env.h │ │ │ ├── models.h │ │ │ ├── nccl │ │ │ ├── test_base.h │ │ │ └── test_utils.h │ │ │ └── util.h │ ├── ir │ │ ├── builder_test.cc │ │ └── ir_test.cc │ ├── session │ │ └── session_test.cc │ ├── test_files │ │ ├── AITOp │ │ │ ├── bmm_permute_a100.so │ │ │ ├── bmm_permute_entry.mlir │ │ │ ├── permute_a100.so │ │ │ └── permute_entry.mlir │ │ ├── Distributed │ │ │ ├── add_send.mlir │ │ │ ├── all_gather.mlir │ │ │ ├── all_reduce.mlir │ │ │ ├── broadcast.mlir │ │ │ ├── broadcast2.mlir │ │ │ ├── ccl.mlir │ │ │ ├── ccl.ptx │ │ │ ├── recv.mlir │ │ │ ├── recv_add.mlir │ │ │ └── send.mlir │ │ ├── DynamicShapes │ │ │ └── Add2 │ │ │ │ ├── entry.mlir │ │ │ │ └── shape_fn.ll │ │ ├── LLJIT │ │ │ ├── Case0 │ │ │ │ ├── entry.mlir │ │ │ │ └── host_kernels.ll │ │ │ ├── Case0_v1_0_0 │ │ │ │ ├── entry.mlirbc │ │ │ │ └── host_kernels.bc │ │ │ ├── add.ll │ │ │ ├── tanh.ll │ │ │ ├── transpose_32_64_64.ll │ │ │ ├── transpose_3_224_224.ll │ │ │ └── typecvt.ll │ │ ├── add2_cpu.mlir │ │ ├── add_splat_const_one_cuda.mlir │ │ ├── cuda_add.cu │ │ ├── custom_add_cpu2cuda.mlir │ │ ├── fill_cuda.mlir │ │ ├── flash_attn_bwd.mlir │ │ ├── flash_attn_bwd_outputs_dk.data │ │ ├── flash_attn_bwd_outputs_dq.data │ │ ├── flash_attn_bwd_outputs_dv.data │ │ ├── flash_attn_fwd.mlir │ │ ├── flash_attn_fwd_outputs.data │ │ ├── flash_attn_inputs_dout.data │ │ ├── flash_attn_inputs_k.data │ │ ├── flash_attn_inputs_q.data │ │ ├── flash_attn_inputs_v.data │ │ ├── flash_attn_kvcache.mlir │ │ ├── flash_attn_kvcache_inputs_cache_seqlens.data │ │ ├── flash_attn_kvcache_inputs_k.data │ │ ├── flash_attn_kvcache_inputs_kcache.data │ │ ├── flash_attn_kvcache_inputs_q.data │ │ ├── flash_attn_kvcache_inputs_v.data │ │ ├── flash_attn_kvcache_inputs_vcache.data │ │ ├── flash_attn_kvcache_outputs.data │ │ ├── flash_attn_kvcache_outputs_kcache.data │ │ ├── flash_attn_kvcache_outputs_vcache.data │ │ ├── generate_flash_attn_ground_truth.py │ │ ├── group_allocation_hook_cpu_group.mlir │ │ ├── llvm_ptx_add.ptx │ │ ├── llvm_ptx_add_bare_ptr.ptx │ │ ├── nvcc_ptx_add.ptx │ │ ├── resnet18_bw_device.ptx │ │ ├── resnet18_bw_host_cuda.mlir │ │ ├── resnet18_fw_bw_device.ptx │ │ ├── resnet18_fw_bw_host_cuda.mlir │ │ ├── resnet18_fw_device.ptx │ │ ├── resnet18_fw_host_cuda.mlir │ │ ├── rng_cuda.mlir │ │ ├── rng_state_cpu.mlir │ │ ├── rng_state_cuda.mlir │ │ ├── string_equal.mlir │ │ └── string_equal_scalar.mlir │ └── unittest_main │ │ └── test_main.cc └── version.ld ├── scripts ├── apply_patches.sh ├── clang_format_check.sh ├── compiler │ └── build_and_test.sh ├── format_check.py ├── prepare.sh └── runtime │ ├── build_and_test.sh │ └── build_external_project.sh ├── talks ├── ChinaSoftCon-ByteIR.pdf └── c4ml23_poster.pdf └── tests ├── build_and_test_e2e.sh ├── compatibility_test ├── execute.py ├── main.py └── reporting.py └── numerical_test ├── execute.py ├── gen_brt_tests.py ├── main.py ├── mlir_tests ├── cpu_ops │ ├── add.mlir │ ├── batch_norm_inference.mlir │ ├── batch_norm_inference_f16.mlir │ ├── broadcast_in_dim.mlir │ ├── compare_LT_f32.mlir │ ├── compare_LT_f64.mlir │ ├── compare_LT_i32.mlir │ ├── compare_LT_i64.mlir │ ├── compare_NE_f32.mlir │ ├── compare_NE_f64.mlir │ ├── compare_NE_i32.mlir │ ├── compare_NE_i64.mlir │ ├── concatenate.mlir │ ├── convert_f16_f32.mlir │ ├── convert_f16_f64.mlir │ ├── convert_f16_i16.mlir │ ├── convert_f16_i32.mlir │ ├── convert_f16_i64.mlir │ ├── convert_f32_f16.mlir │ ├── convert_f32_f64.mlir │ ├── convert_f32_i16.mlir │ ├── convert_f32_i32.mlir │ ├── convert_f32_i32_special_val.mlir │ ├── convert_f32_i64.mlir │ ├── convert_f64_f16.mlir │ ├── convert_f64_f32.mlir │ ├── convert_f64_i16.mlir │ ├── convert_f64_i32.mlir │ ├── convert_f64_i64.mlir │ ├── convert_i16_f16.mlir │ ├── convert_i16_f32.mlir │ ├── convert_i16_f64.mlir │ ├── convert_i16_i32.mlir │ ├── convert_i16_i64.mlir │ ├── convert_i32_f16.mlir │ ├── convert_i32_f32.mlir │ ├── convert_i32_f64.mlir │ ├── convert_i32_i16.mlir │ ├── convert_i32_i64.mlir │ ├── convert_i64_f16.mlir │ ├── convert_i64_f32.mlir │ ├── convert_i64_f64.mlir │ ├── convert_i64_i16.mlir │ ├── convert_i64_i32.mlir │ ├── custom_call_byteir_addn.mlir │ ├── custom_call_byteir_arg_max.mlir │ ├── custom_call_byteir_arg_max_i32.mlir │ ├── custom_call_byteir_arg_min.mlir │ ├── custom_call_byteir_arg_min_i32.mlir │ ├── custom_call_byteir_l2norm.mlir │ ├── custom_call_byteir_softmax.mlir │ ├── custom_call_tf_UpperBound.mlir │ ├── divide_f16.mlir │ ├── log_plus_one_f16.mlir │ ├── maximum_f32.mlir │ ├── maximum_f64.mlir │ ├── maximum_i32.mlir │ ├── maximum_i64.mlir │ ├── minimum_f32.mlir │ ├── minimum_f64.mlir │ ├── minimum_i32.mlir │ ├── minimum_i64.mlir │ ├── multiply_f32.mlir │ ├── multiply_f64.mlir │ ├── multiply_i32.mlir │ ├── multiply_i64.mlir │ ├── reduce_f32.mlir │ ├── remainder_i64.mlir │ ├── reshape_slice.mlir │ ├── rng.mlir │ ├── scatter_insert_slice.mlir │ ├── select_f32.mlir │ ├── select_f64.mlir │ ├── select_i64.mlir │ ├── slice_view_like.mlir │ └── subtrace_f16.mlir └── ops │ ├── add.mlir │ ├── bmm_rcr.mlir │ ├── bmm_rrc.mlir │ ├── bmm_rrr_add_f16.mlir │ ├── bmm_rrr_f16.mlir │ ├── bmm_rrr_permute_f16.mlir │ ├── bmm_rrr_permute_f32.mlir │ ├── broadcast.mlir │ ├── broadcast1.mlir │ ├── compare_eq.mlir │ ├── compare_lt.mlir │ ├── concat.mlir │ ├── concat2.mlir │ ├── convert_f16_f32.mlir │ ├── convert_f32_f16.mlir │ ├── divide.mlir │ ├── gather.mlir │ ├── gemm_crr_f16.mlir │ ├── gemm_rrr_f16.mlir │ ├── gemm_rrr_f32.mlir │ ├── insert_slice.mlir │ ├── layernorm.mlir │ ├── logistic.mlir │ ├── mul_f16.mlir │ ├── mul_f32.mlir │ ├── negate.mlir │ ├── power.mlir │ ├── reduce_first_dim.mlir │ ├── reduce_sum.mlir │ ├── reduce_sum_2d.mlir │ ├── reduce_sum_first_2d.mlir │ ├── rsqrt.mlir │ ├── scatter.mlir │ ├── scatter_insert_slice.mlir │ ├── select.mlir │ ├── slice.mlir │ ├── softmax.mlir │ ├── transpose0312.mlir │ ├── transpose102.mlir │ ├── transpose1023.mlir │ ├── transpose120.mlir │ ├── transpose1203.mlir │ ├── transpose2013.mlir │ └── transpose2d.mlir ├── profiler.py ├── reporting.py ├── testset.py ├── torch_dynamo_e2e_testing ├── backend.py ├── execute.py └── test_suite │ └── test_flash_attn.py └── torch_e2e_testing ├── framework.py ├── registry.py └── test_suite ├── __init__.py └── basic.py /.clang-format: -------------------------------------------------------------------------------- 1 | BasedOnStyle: LLVM 2 | -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/bug.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/.github/ISSUE_TEMPLATE/bug.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature-request.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/.github/ISSUE_TEMPLATE/feature-request.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/question.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/.github/ISSUE_TEMPLATE/question.md -------------------------------------------------------------------------------- /.github/workflows/compiler-ci.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/.github/workflows/compiler-ci.yaml -------------------------------------------------------------------------------- /.github/workflows/daily_ci.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/.github/workflows/daily_ci.yaml -------------------------------------------------------------------------------- /.github/workflows/e2e_test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/.github/workflows/e2e_test.yaml -------------------------------------------------------------------------------- /.github/workflows/format-check.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/.github/workflows/format-check.yaml -------------------------------------------------------------------------------- /.github/workflows/onnx-frontend-ci.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/.github/workflows/onnx-frontend-ci.yaml -------------------------------------------------------------------------------- /.github/workflows/runtime-ci.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/.github/workflows/runtime-ci.yaml -------------------------------------------------------------------------------- /.github/workflows/tf-frontend-ci.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/.github/workflows/tf-frontend-ci.yaml -------------------------------------------------------------------------------- /.github/workflows/torch-frontend-ci.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/.github/workflows/torch-frontend-ci.yaml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/.gitmodules -------------------------------------------------------------------------------- /CITATION.cff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/CITATION.cff -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/LICENSE -------------------------------------------------------------------------------- /NOTICE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/NOTICE -------------------------------------------------------------------------------- /README-zh_cn.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/README-zh_cn.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/README.md -------------------------------------------------------------------------------- /compiler/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/.gitignore -------------------------------------------------------------------------------- /compiler/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/README.md -------------------------------------------------------------------------------- /compiler/cmake/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/cmake/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/cmake/MLIR.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/cmake/MLIR.cmake -------------------------------------------------------------------------------- /compiler/cmake/mhlo.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/cmake/mhlo.cmake -------------------------------------------------------------------------------- /compiler/dialects/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/dialects/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/dialects/include/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(byteir) -------------------------------------------------------------------------------- /compiler/dialects/include/byteir/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(Dialect) -------------------------------------------------------------------------------- /compiler/dialects/include/byteir/Dialect/Ccl/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(IR) -------------------------------------------------------------------------------- /compiler/dialects/lib/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(Dialect) -------------------------------------------------------------------------------- /compiler/dialects/lib/Dialect/Ace/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/dialects/lib/Dialect/Ace/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/dialects/lib/Dialect/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/dialects/lib/Dialect/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/dialects/lib/Dialect/Ccl/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(IR) -------------------------------------------------------------------------------- /compiler/dialects/lib/Dialect/Ccl/IR/CclOps.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/dialects/lib/Dialect/Ccl/IR/CclOps.cpp -------------------------------------------------------------------------------- /compiler/doc/attention.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/doc/attention.md -------------------------------------------------------------------------------- /compiler/doc/byteir_hlo_custom_call.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/doc/byteir_hlo_custom_call.md -------------------------------------------------------------------------------- /compiler/doc/codegen.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/doc/codegen.md -------------------------------------------------------------------------------- /compiler/doc/gpu.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/doc/gpu.md -------------------------------------------------------------------------------- /compiler/doc/linalg.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/doc/linalg.md -------------------------------------------------------------------------------- /compiler/doc/passes.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/doc/passes.md -------------------------------------------------------------------------------- /compiler/doc/rng.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/doc/rng.md -------------------------------------------------------------------------------- /compiler/include/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(byteir) 2 | -------------------------------------------------------------------------------- /compiler/include/byteir-c/Dialects.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir-c/Dialects.h -------------------------------------------------------------------------------- /compiler/include/byteir-c/PDLValue.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir-c/PDLValue.h -------------------------------------------------------------------------------- /compiler/include/byteir-c/Passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir-c/Passes.h -------------------------------------------------------------------------------- /compiler/include/byteir-c/Translation.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir-c/Translation.h -------------------------------------------------------------------------------- /compiler/include/byteir/Analysis/Alias.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Analysis/Alias.h -------------------------------------------------------------------------------- /compiler/include/byteir/Analysis/DimFlag.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Analysis/DimFlag.h -------------------------------------------------------------------------------- /compiler/include/byteir/Analysis/Liveness.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Analysis/Liveness.h -------------------------------------------------------------------------------- /compiler/include/byteir/Analysis/OpDependence.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Analysis/OpDependence.h -------------------------------------------------------------------------------- /compiler/include/byteir/Analysis/ShapeAnalysis.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Analysis/ShapeAnalysis.h -------------------------------------------------------------------------------- /compiler/include/byteir/Analysis/SideEffect.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Analysis/SideEffect.h -------------------------------------------------------------------------------- /compiler/include/byteir/Analysis/SymbolicShape.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Analysis/SymbolicShape.h -------------------------------------------------------------------------------- /compiler/include/byteir/Analysis/UseRange.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Analysis/UseRange.h -------------------------------------------------------------------------------- /compiler/include/byteir/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/include/byteir/Conversion/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Conversion/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/include/byteir/Conversion/Passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Conversion/Passes.h -------------------------------------------------------------------------------- /compiler/include/byteir/Conversion/Passes.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Conversion/Passes.td -------------------------------------------------------------------------------- /compiler/include/byteir/Conversion/ToAIT/ToAIT.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Conversion/ToAIT/ToAIT.h -------------------------------------------------------------------------------- /compiler/include/byteir/Conversion/ToByre/ToByre.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Conversion/ToByre/ToByre.h -------------------------------------------------------------------------------- /compiler/include/byteir/Conversion/ToGPU/ToGPU.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Conversion/ToGPU/ToGPU.h -------------------------------------------------------------------------------- /compiler/include/byteir/Conversion/ToGPU/Utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Conversion/ToGPU/Utils.h -------------------------------------------------------------------------------- /compiler/include/byteir/Conversion/ToPTX/ToPTX.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Conversion/ToPTX/ToPTX.h -------------------------------------------------------------------------------- /compiler/include/byteir/Conversion/ToTIT/ToTIT.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Conversion/ToTIT/ToTIT.h -------------------------------------------------------------------------------- /compiler/include/byteir/Dialect/Ace/Passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Dialect/Ace/Passes.h -------------------------------------------------------------------------------- /compiler/include/byteir/Dialect/Ace/Passes.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Dialect/Ace/Passes.td -------------------------------------------------------------------------------- /compiler/include/byteir/Dialect/Affine/Passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Dialect/Affine/Passes.h -------------------------------------------------------------------------------- /compiler/include/byteir/Dialect/Affine/Passes.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Dialect/Affine/Passes.td -------------------------------------------------------------------------------- /compiler/include/byteir/Dialect/Byre/ByreBase.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Dialect/Byre/ByreBase.td -------------------------------------------------------------------------------- /compiler/include/byteir/Dialect/Byre/ByreOps.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Dialect/Byre/ByreOps.td -------------------------------------------------------------------------------- /compiler/include/byteir/Dialect/Byre/Common.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Dialect/Byre/Common.h -------------------------------------------------------------------------------- /compiler/include/byteir/Dialect/Byre/Passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Dialect/Byre/Passes.h -------------------------------------------------------------------------------- /compiler/include/byteir/Dialect/Byre/Passes.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Dialect/Byre/Passes.td -------------------------------------------------------------------------------- /compiler/include/byteir/Dialect/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Dialect/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/include/byteir/Dialect/Cat/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(IR) 2 | -------------------------------------------------------------------------------- /compiler/include/byteir/Dialect/Cat/IR/CatOps.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Dialect/Cat/IR/CatOps.td -------------------------------------------------------------------------------- /compiler/include/byteir/Dialect/Ccl/Passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Dialect/Ccl/Passes.h -------------------------------------------------------------------------------- /compiler/include/byteir/Dialect/Ccl/Passes.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Dialect/Ccl/Passes.td -------------------------------------------------------------------------------- /compiler/include/byteir/Dialect/GPU/Passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Dialect/GPU/Passes.h -------------------------------------------------------------------------------- /compiler/include/byteir/Dialect/GPU/Passes.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Dialect/GPU/Passes.td -------------------------------------------------------------------------------- /compiler/include/byteir/Dialect/Lace/LaceBase.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Dialect/Lace/LaceBase.td -------------------------------------------------------------------------------- /compiler/include/byteir/Dialect/Lace/LaceOps.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Dialect/Lace/LaceOps.td -------------------------------------------------------------------------------- /compiler/include/byteir/Dialect/Lccl/LcclBase.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Dialect/Lccl/LcclBase.td -------------------------------------------------------------------------------- /compiler/include/byteir/Dialect/Lccl/LcclOps.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Dialect/Lccl/LcclOps.h -------------------------------------------------------------------------------- /compiler/include/byteir/Dialect/Lccl/LcclOps.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Dialect/Lccl/LcclOps.td -------------------------------------------------------------------------------- /compiler/include/byteir/Dialect/Linalg/Passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Dialect/Linalg/Passes.h -------------------------------------------------------------------------------- /compiler/include/byteir/Dialect/Linalg/Passes.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Dialect/Linalg/Passes.td -------------------------------------------------------------------------------- /compiler/include/byteir/Dialect/MemRef/Passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Dialect/MemRef/Passes.h -------------------------------------------------------------------------------- /compiler/include/byteir/Dialect/MemRef/Passes.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Dialect/MemRef/Passes.td -------------------------------------------------------------------------------- /compiler/include/byteir/Dialect/SCF/Passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Dialect/SCF/Passes.h -------------------------------------------------------------------------------- /compiler/include/byteir/Dialect/SCF/Passes.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Dialect/SCF/Passes.td -------------------------------------------------------------------------------- /compiler/include/byteir/Dialect/SCF/Util/Util.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Dialect/SCF/Util/Util.h -------------------------------------------------------------------------------- /compiler/include/byteir/Dialect/Shape/Passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Dialect/Shape/Passes.h -------------------------------------------------------------------------------- /compiler/include/byteir/Dialect/Shape/Passes.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Dialect/Shape/Passes.td -------------------------------------------------------------------------------- /compiler/include/byteir/Dialect/Tensor/Passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Dialect/Tensor/Passes.h -------------------------------------------------------------------------------- /compiler/include/byteir/Dialect/Tensor/Passes.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Dialect/Tensor/Passes.td -------------------------------------------------------------------------------- /compiler/include/byteir/Dialect/Vector/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(Transforms) -------------------------------------------------------------------------------- /compiler/include/byteir/Dialect/mhlo/Passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Dialect/mhlo/Passes.h -------------------------------------------------------------------------------- /compiler/include/byteir/Dialect/mhlo/Passes.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Dialect/mhlo/Passes.td -------------------------------------------------------------------------------- /compiler/include/byteir/Dialect/mhlo/Util/Util.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Dialect/mhlo/Util/Util.h -------------------------------------------------------------------------------- /compiler/include/byteir/Pipelines/AffineOpt.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Pipelines/AffineOpt.h -------------------------------------------------------------------------------- /compiler/include/byteir/Pipelines/AllOpt.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Pipelines/AllOpt.h -------------------------------------------------------------------------------- /compiler/include/byteir/Pipelines/BufferizeOpt.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Pipelines/BufferizeOpt.h -------------------------------------------------------------------------------- /compiler/include/byteir/Pipelines/ByreHost.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Pipelines/ByreHost.h -------------------------------------------------------------------------------- /compiler/include/byteir/Pipelines/ByreOpt.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Pipelines/ByreOpt.h -------------------------------------------------------------------------------- /compiler/include/byteir/Pipelines/CatFusionOpt.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Pipelines/CatFusionOpt.h -------------------------------------------------------------------------------- /compiler/include/byteir/Pipelines/Common/Utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Pipelines/Common/Utils.h -------------------------------------------------------------------------------- /compiler/include/byteir/Pipelines/GPU/GPUOpt.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Pipelines/GPU/GPUOpt.h -------------------------------------------------------------------------------- /compiler/include/byteir/Pipelines/HloFusionOpt.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Pipelines/HloFusionOpt.h -------------------------------------------------------------------------------- /compiler/include/byteir/Pipelines/HloGraphOpt.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Pipelines/HloGraphOpt.h -------------------------------------------------------------------------------- /compiler/include/byteir/Pipelines/Host/Codegen.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Pipelines/Host/Codegen.h -------------------------------------------------------------------------------- /compiler/include/byteir/Pipelines/Host/HostOpt.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Pipelines/Host/HostOpt.h -------------------------------------------------------------------------------- /compiler/include/byteir/Pipelines/Host/ToLLVM.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Pipelines/Host/ToLLVM.h -------------------------------------------------------------------------------- /compiler/include/byteir/Pipelines/SCFOpt.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Pipelines/SCFOpt.h -------------------------------------------------------------------------------- /compiler/include/byteir/Pipelines/ShapeOpt.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Pipelines/ShapeOpt.h -------------------------------------------------------------------------------- /compiler/include/byteir/Stat/AllocCnt/AllocCnt.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Stat/AllocCnt/AllocCnt.h -------------------------------------------------------------------------------- /compiler/include/byteir/Stat/Common/Reg.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Stat/Common/Reg.h -------------------------------------------------------------------------------- /compiler/include/byteir/Stat/InitAllStats.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Stat/InitAllStats.h -------------------------------------------------------------------------------- /compiler/include/byteir/Stat/OpCnt/OpCnt.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Stat/OpCnt/OpCnt.h -------------------------------------------------------------------------------- /compiler/include/byteir/Target/CUDA/ToCUDA.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Target/CUDA/ToCUDA.h -------------------------------------------------------------------------------- /compiler/include/byteir/Target/Common/Common.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Target/Common/Common.h -------------------------------------------------------------------------------- /compiler/include/byteir/Target/Common/EmitUtil.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Target/Common/EmitUtil.h -------------------------------------------------------------------------------- /compiler/include/byteir/Target/Cpp/CppEmitter.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Target/Cpp/CppEmitter.h -------------------------------------------------------------------------------- /compiler/include/byteir/Target/Cpp/ToCpp.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Target/Cpp/ToCpp.h -------------------------------------------------------------------------------- /compiler/include/byteir/Target/LLVM/ToLLVMBC.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Target/LLVM/ToLLVMBC.h -------------------------------------------------------------------------------- /compiler/include/byteir/Target/PTX/Passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Target/PTX/Passes.h -------------------------------------------------------------------------------- /compiler/include/byteir/Target/PTX/ToPTX.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Target/PTX/ToPTX.h -------------------------------------------------------------------------------- /compiler/include/byteir/Transforms/Bufferize.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Transforms/Bufferize.h -------------------------------------------------------------------------------- /compiler/include/byteir/Transforms/CMAE.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Transforms/CMAE.h -------------------------------------------------------------------------------- /compiler/include/byteir/Transforms/CollectFunc.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Transforms/CollectFunc.h -------------------------------------------------------------------------------- /compiler/include/byteir/Transforms/FuncTag.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Transforms/FuncTag.h -------------------------------------------------------------------------------- /compiler/include/byteir/Transforms/LoopTag.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Transforms/LoopTag.h -------------------------------------------------------------------------------- /compiler/include/byteir/Transforms/LoopUnroll.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Transforms/LoopUnroll.h -------------------------------------------------------------------------------- /compiler/include/byteir/Transforms/ModuleTag.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Transforms/ModuleTag.h -------------------------------------------------------------------------------- /compiler/include/byteir/Transforms/Passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Transforms/Passes.h -------------------------------------------------------------------------------- /compiler/include/byteir/Transforms/Passes.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Transforms/Passes.td -------------------------------------------------------------------------------- /compiler/include/byteir/Transforms/SetArgShape.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Transforms/SetArgShape.h -------------------------------------------------------------------------------- /compiler/include/byteir/Transforms/SetSpace.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Transforms/SetSpace.h -------------------------------------------------------------------------------- /compiler/include/byteir/Transforms/Utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Transforms/Utils.h -------------------------------------------------------------------------------- /compiler/include/byteir/Utils/AffineUtils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Utils/AffineUtils.h -------------------------------------------------------------------------------- /compiler/include/byteir/Utils/AttrUtils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Utils/AttrUtils.h -------------------------------------------------------------------------------- /compiler/include/byteir/Utils/FuncUtils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Utils/FuncUtils.h -------------------------------------------------------------------------------- /compiler/include/byteir/Utils/GraphUtils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Utils/GraphUtils.h -------------------------------------------------------------------------------- /compiler/include/byteir/Utils/HashUtils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Utils/HashUtils.h -------------------------------------------------------------------------------- /compiler/include/byteir/Utils/Hoist.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Utils/Hoist.h -------------------------------------------------------------------------------- /compiler/include/byteir/Utils/IRRewrite.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Utils/IRRewrite.h -------------------------------------------------------------------------------- /compiler/include/byteir/Utils/LoopUtils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Utils/LoopUtils.h -------------------------------------------------------------------------------- /compiler/include/byteir/Utils/MemUtils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Utils/MemUtils.h -------------------------------------------------------------------------------- /compiler/include/byteir/Utils/ModuleUtils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Utils/ModuleUtils.h -------------------------------------------------------------------------------- /compiler/include/byteir/Utils/OpInterfaceUtils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Utils/OpInterfaceUtils.h -------------------------------------------------------------------------------- /compiler/include/byteir/Utils/OptionUtils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Utils/OptionUtils.h -------------------------------------------------------------------------------- /compiler/include/byteir/Utils/PatternMatch.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Utils/PatternMatch.h -------------------------------------------------------------------------------- /compiler/include/byteir/Utils/PipelineUtils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Utils/PipelineUtils.h -------------------------------------------------------------------------------- /compiler/include/byteir/Utils/TileUtils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Utils/TileUtils.h -------------------------------------------------------------------------------- /compiler/include/byteir/Utils/TypeUtils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Utils/TypeUtils.h -------------------------------------------------------------------------------- /compiler/include/byteir/Utils/Utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/include/byteir/Utils/Utils.h -------------------------------------------------------------------------------- /compiler/lib/Analysis/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Analysis/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Analysis/DimFlag.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Analysis/DimFlag.cpp -------------------------------------------------------------------------------- /compiler/lib/Analysis/Liveness.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Analysis/Liveness.cpp -------------------------------------------------------------------------------- /compiler/lib/Analysis/OpDependence.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Analysis/OpDependence.cpp -------------------------------------------------------------------------------- /compiler/lib/Analysis/ShapeAnalysis.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Analysis/ShapeAnalysis.cpp -------------------------------------------------------------------------------- /compiler/lib/Analysis/SideEffect.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Analysis/SideEffect.cpp -------------------------------------------------------------------------------- /compiler/lib/Analysis/SymbolicShape.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Analysis/SymbolicShape.cpp -------------------------------------------------------------------------------- /compiler/lib/Analysis/UseRange.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Analysis/UseRange.cpp -------------------------------------------------------------------------------- /compiler/lib/CAPI/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/CAPI/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/CAPI/Dialects.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/CAPI/Dialects.cpp -------------------------------------------------------------------------------- /compiler/lib/CAPI/PDLValue.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/CAPI/PDLValue.cpp -------------------------------------------------------------------------------- /compiler/lib/CAPI/Passes.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/CAPI/Passes.cpp -------------------------------------------------------------------------------- /compiler/lib/CAPI/Translation.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/CAPI/Translation.cpp -------------------------------------------------------------------------------- /compiler/lib/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Conversion/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Conversion/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Conversion/Common/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Conversion/Common/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Conversion/GPUToNVVM/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Conversion/GPUToNVVM/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Conversion/GPUToNVVM/GPUToNVVM.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Conversion/GPUToNVVM/GPUToNVVM.cpp -------------------------------------------------------------------------------- /compiler/lib/Conversion/HloToCat/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Conversion/HloToCat/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Conversion/HloToCat/HloToCat.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Conversion/HloToCat/HloToCat.cpp -------------------------------------------------------------------------------- /compiler/lib/Conversion/HloToCat/Utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Conversion/HloToCat/Utils.h -------------------------------------------------------------------------------- /compiler/lib/Conversion/PassDetail.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Conversion/PassDetail.h -------------------------------------------------------------------------------- /compiler/lib/Conversion/ToAIT/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Conversion/ToAIT/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Conversion/ToAIT/GenAITConfig.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Conversion/ToAIT/GenAITConfig.cpp -------------------------------------------------------------------------------- /compiler/lib/Conversion/ToAce/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Conversion/ToAce/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Conversion/ToAce/MhloToAce.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Conversion/ToAce/MhloToAce.cpp -------------------------------------------------------------------------------- /compiler/lib/Conversion/ToByre/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Conversion/ToByre/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Conversion/ToByre/ToByre.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Conversion/ToByre/ToByre.cpp -------------------------------------------------------------------------------- /compiler/lib/Conversion/ToGPU/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Conversion/ToGPU/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Conversion/ToGPU/FuncToGPU.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Conversion/ToGPU/FuncToGPU.cpp -------------------------------------------------------------------------------- /compiler/lib/Conversion/ToGPU/Utils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Conversion/ToGPU/Utils.cpp -------------------------------------------------------------------------------- /compiler/lib/Conversion/ToHlo/ArithToMhlo.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Conversion/ToHlo/ArithToMhlo.cpp -------------------------------------------------------------------------------- /compiler/lib/Conversion/ToHlo/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Conversion/ToHlo/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Conversion/ToLLVM/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Conversion/ToLLVM/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Conversion/ToLLVM/GenLLVMConfig.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Conversion/ToLLVM/GenLLVMConfig.cpp -------------------------------------------------------------------------------- /compiler/lib/Conversion/ToLinalg/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Conversion/ToLinalg/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Conversion/ToLinalg/HloToLinalg.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Conversion/ToLinalg/HloToLinalg.cpp -------------------------------------------------------------------------------- /compiler/lib/Conversion/ToPTX/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Conversion/ToPTX/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Conversion/ToPTX/GenPTXConfig.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Conversion/ToPTX/GenPTXConfig.cpp -------------------------------------------------------------------------------- /compiler/lib/Conversion/ToTIT/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Conversion/ToTIT/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Conversion/ToTIT/GenTITConfig.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Conversion/ToTIT/GenTITConfig.cpp -------------------------------------------------------------------------------- /compiler/lib/Dialect/Ace/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/Ace/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Dialect/Ace/Transforms/PassDetail.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/Ace/Transforms/PassDetail.h -------------------------------------------------------------------------------- /compiler/lib/Dialect/Affine/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/Affine/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Dialect/Byre/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/Byre/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Dialect/Byre/IR/ByreDialect.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/Byre/IR/ByreDialect.cpp -------------------------------------------------------------------------------- /compiler/lib/Dialect/Byre/IR/Common.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/Byre/IR/Common.cpp -------------------------------------------------------------------------------- /compiler/lib/Dialect/Byre/IR/Serialization.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/Byre/IR/Serialization.cpp -------------------------------------------------------------------------------- /compiler/lib/Dialect/Byre/Transforms/Serial.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/Byre/Transforms/Serial.cpp -------------------------------------------------------------------------------- /compiler/lib/Dialect/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Dialect/Cat/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(IR) -------------------------------------------------------------------------------- /compiler/lib/Dialect/Cat/IR/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/Cat/IR/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Dialect/Cat/IR/CatDialect.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/Cat/IR/CatDialect.cpp -------------------------------------------------------------------------------- /compiler/lib/Dialect/Ccl/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/Ccl/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Dialect/Ccl/Transforms/PassDetail.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/Ccl/Transforms/PassDetail.h -------------------------------------------------------------------------------- /compiler/lib/Dialect/GPU/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/GPU/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Dialect/GPU/TransformOps/Utils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/GPU/TransformOps/Utils.cpp -------------------------------------------------------------------------------- /compiler/lib/Dialect/GPU/Transforms/PassDetail.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/GPU/Transforms/PassDetail.h -------------------------------------------------------------------------------- /compiler/lib/Dialect/GPU/Transforms/Utils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/GPU/Transforms/Utils.cpp -------------------------------------------------------------------------------- /compiler/lib/Dialect/Lace/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/Lace/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Dialect/Lace/IR/LaceDialect.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/Lace/IR/LaceDialect.cpp -------------------------------------------------------------------------------- /compiler/lib/Dialect/Lccl/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(IR) 2 | -------------------------------------------------------------------------------- /compiler/lib/Dialect/Lccl/IR/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/Lccl/IR/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Dialect/Lccl/IR/LcclOps.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/Lccl/IR/LcclOps.cpp -------------------------------------------------------------------------------- /compiler/lib/Dialect/Linalg/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/Linalg/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Dialect/Linalg/IR/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/Linalg/IR/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Dialect/Linalg/IR/LinalgExtOps.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/Linalg/IR/LinalgExtOps.cpp -------------------------------------------------------------------------------- /compiler/lib/Dialect/Linalg/Util/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/Linalg/Util/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Dialect/Linalg/Util/Util.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/Linalg/Util/Util.cpp -------------------------------------------------------------------------------- /compiler/lib/Dialect/MemRef/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/MemRef/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Dialect/MemRef/Utils/Layout.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/MemRef/Utils/Layout.cpp -------------------------------------------------------------------------------- /compiler/lib/Dialect/MemRef/Utils/MemEffect.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/MemRef/Utils/MemEffect.cpp -------------------------------------------------------------------------------- /compiler/lib/Dialect/MemRef/Utils/Ops.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/MemRef/Utils/Ops.cpp -------------------------------------------------------------------------------- /compiler/lib/Dialect/SCF/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/SCF/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Dialect/SCF/Transforms/PassDetail.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/SCF/Transforms/PassDetail.h -------------------------------------------------------------------------------- /compiler/lib/Dialect/SCF/Util/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/SCF/Util/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Dialect/SCF/Util/Util.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/SCF/Util/Util.cpp -------------------------------------------------------------------------------- /compiler/lib/Dialect/Shape/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/Shape/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Dialect/Shape/IR/ShapeExtOps.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/Shape/IR/ShapeExtOps.cpp -------------------------------------------------------------------------------- /compiler/lib/Dialect/Tensor/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/Tensor/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Dialect/Tensor/IR/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/Tensor/IR/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Dialect/Transform/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/Transform/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Dialect/Transform/IR/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/Transform/IR/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Dialect/Vector/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(Transforms) 2 | -------------------------------------------------------------------------------- /compiler/lib/Dialect/mhlo/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/mhlo/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Dialect/mhlo/Util/FusionUtil.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/mhlo/Util/FusionUtil.cpp -------------------------------------------------------------------------------- /compiler/lib/Dialect/mhlo/Util/Util.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Dialect/mhlo/Util/Util.cpp -------------------------------------------------------------------------------- /compiler/lib/Pipelines/AffineOpt.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Pipelines/AffineOpt.cpp -------------------------------------------------------------------------------- /compiler/lib/Pipelines/AllOpt.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Pipelines/AllOpt.cpp -------------------------------------------------------------------------------- /compiler/lib/Pipelines/BufferizeOpt.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Pipelines/BufferizeOpt.cpp -------------------------------------------------------------------------------- /compiler/lib/Pipelines/ByreHost.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Pipelines/ByreHost.cpp -------------------------------------------------------------------------------- /compiler/lib/Pipelines/ByreOpt.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Pipelines/ByreOpt.cpp -------------------------------------------------------------------------------- /compiler/lib/Pipelines/ByreTensorOpt.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Pipelines/ByreTensorOpt.cpp -------------------------------------------------------------------------------- /compiler/lib/Pipelines/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Pipelines/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Pipelines/CatFusionOpt.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Pipelines/CatFusionOpt.cpp -------------------------------------------------------------------------------- /compiler/lib/Pipelines/CatPreprocess.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Pipelines/CatPreprocess.cpp -------------------------------------------------------------------------------- /compiler/lib/Pipelines/Common/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Pipelines/Common/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Pipelines/Common/Utils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Pipelines/Common/Utils.cpp -------------------------------------------------------------------------------- /compiler/lib/Pipelines/GPU/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Pipelines/GPU/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Pipelines/GPU/GPUOpt.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Pipelines/GPU/GPUOpt.cpp -------------------------------------------------------------------------------- /compiler/lib/Pipelines/GPU/LinalgMemrefGPU.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Pipelines/GPU/LinalgMemrefGPU.cpp -------------------------------------------------------------------------------- /compiler/lib/Pipelines/GPU/MappingForall.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Pipelines/GPU/MappingForall.cpp -------------------------------------------------------------------------------- /compiler/lib/Pipelines/GPU/NVVMCodegen.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Pipelines/GPU/NVVMCodegen.cpp -------------------------------------------------------------------------------- /compiler/lib/Pipelines/GPU/ReductionCodegen.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Pipelines/GPU/ReductionCodegen.cpp -------------------------------------------------------------------------------- /compiler/lib/Pipelines/HloFusionOpt.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Pipelines/HloFusionOpt.cpp -------------------------------------------------------------------------------- /compiler/lib/Pipelines/HloGraphOpt.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Pipelines/HloGraphOpt.cpp -------------------------------------------------------------------------------- /compiler/lib/Pipelines/Host/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Pipelines/Host/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Pipelines/Host/Codegen.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Pipelines/Host/Codegen.cpp -------------------------------------------------------------------------------- /compiler/lib/Pipelines/Host/HostOpt.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Pipelines/Host/HostOpt.cpp -------------------------------------------------------------------------------- /compiler/lib/Pipelines/Host/ToLLVM.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Pipelines/Host/ToLLVM.cpp -------------------------------------------------------------------------------- /compiler/lib/Pipelines/LinalgMemrefOpt.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Pipelines/LinalgMemrefOpt.cpp -------------------------------------------------------------------------------- /compiler/lib/Pipelines/LinalgTensorOpt.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Pipelines/LinalgTensorOpt.cpp -------------------------------------------------------------------------------- /compiler/lib/Pipelines/SCFOpt.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Pipelines/SCFOpt.cpp -------------------------------------------------------------------------------- /compiler/lib/Pipelines/ShapeOpt.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Pipelines/ShapeOpt.cpp -------------------------------------------------------------------------------- /compiler/lib/Stat/AllocCnt/AllocCnt.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Stat/AllocCnt/AllocCnt.cpp -------------------------------------------------------------------------------- /compiler/lib/Stat/AllocCnt/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Stat/AllocCnt/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Stat/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Stat/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Stat/Common/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Stat/Common/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Stat/Common/Reg.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Stat/Common/Reg.cpp -------------------------------------------------------------------------------- /compiler/lib/Stat/OpCnt/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Stat/OpCnt/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Stat/OpCnt/OpCnt.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Stat/OpCnt/OpCnt.cpp -------------------------------------------------------------------------------- /compiler/lib/Target/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Target/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Target/CUDA/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Target/CUDA/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Target/CUDA/TranslateToCUDA.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Target/CUDA/TranslateToCUDA.cpp -------------------------------------------------------------------------------- /compiler/lib/Target/Cpp/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Target/Cpp/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Target/Cpp/TranslateToCpp.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Target/Cpp/TranslateToCpp.cpp -------------------------------------------------------------------------------- /compiler/lib/Target/LLVM/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Target/LLVM/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Target/PTX/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Target/PTX/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Target/PTX/GPUKernelToPTX.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Target/PTX/GPUKernelToPTX.cpp -------------------------------------------------------------------------------- /compiler/lib/Target/PTX/TranslateToPTX.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Target/PTX/TranslateToPTX.cpp -------------------------------------------------------------------------------- /compiler/lib/Transforms/AnchoredPipeline.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Transforms/AnchoredPipeline.cpp -------------------------------------------------------------------------------- /compiler/lib/Transforms/ApplyPDLPatterns.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Transforms/ApplyPDLPatterns.cpp -------------------------------------------------------------------------------- /compiler/lib/Transforms/Bufferize.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Transforms/Bufferize.cpp -------------------------------------------------------------------------------- /compiler/lib/Transforms/CMAE.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Transforms/CMAE.cpp -------------------------------------------------------------------------------- /compiler/lib/Transforms/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Transforms/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Transforms/CanonicalizeExt.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Transforms/CanonicalizeExt.cpp -------------------------------------------------------------------------------- /compiler/lib/Transforms/CollectFunc.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Transforms/CollectFunc.cpp -------------------------------------------------------------------------------- /compiler/lib/Transforms/CondCanonicalize.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Transforms/CondCanonicalize.cpp -------------------------------------------------------------------------------- /compiler/lib/Transforms/FuncTag.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Transforms/FuncTag.cpp -------------------------------------------------------------------------------- /compiler/lib/Transforms/GenericDeviceConfig.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Transforms/GenericDeviceConfig.cpp -------------------------------------------------------------------------------- /compiler/lib/Transforms/InsertUniqueId.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Transforms/InsertUniqueId.cpp -------------------------------------------------------------------------------- /compiler/lib/Transforms/LoopTag.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Transforms/LoopTag.cpp -------------------------------------------------------------------------------- /compiler/lib/Transforms/LoopUnroll.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Transforms/LoopUnroll.cpp -------------------------------------------------------------------------------- /compiler/lib/Transforms/MemoryPlanning.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Transforms/MemoryPlanning.cpp -------------------------------------------------------------------------------- /compiler/lib/Transforms/ModuleTag.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Transforms/ModuleTag.cpp -------------------------------------------------------------------------------- /compiler/lib/Transforms/PassDetail.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Transforms/PassDetail.h -------------------------------------------------------------------------------- /compiler/lib/Transforms/RemoveFuncBody.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Transforms/RemoveFuncBody.cpp -------------------------------------------------------------------------------- /compiler/lib/Transforms/RewriteOpToStdCall.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Transforms/RewriteOpToStdCall.cpp -------------------------------------------------------------------------------- /compiler/lib/Transforms/SetArgShape.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Transforms/SetArgShape.cpp -------------------------------------------------------------------------------- /compiler/lib/Transforms/SetSpace.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Transforms/SetSpace.cpp -------------------------------------------------------------------------------- /compiler/lib/Transforms/ShapeFuncOutlining.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Transforms/ShapeFuncOutlining.cpp -------------------------------------------------------------------------------- /compiler/lib/Transforms/Utils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Transforms/Utils.cpp -------------------------------------------------------------------------------- /compiler/lib/Utils/AffineUtils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Utils/AffineUtils.cpp -------------------------------------------------------------------------------- /compiler/lib/Utils/AttrUtils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Utils/AttrUtils.cpp -------------------------------------------------------------------------------- /compiler/lib/Utils/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Utils/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/lib/Utils/FuncUtils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Utils/FuncUtils.cpp -------------------------------------------------------------------------------- /compiler/lib/Utils/GraphUtils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Utils/GraphUtils.cpp -------------------------------------------------------------------------------- /compiler/lib/Utils/Hoist.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Utils/Hoist.cpp -------------------------------------------------------------------------------- /compiler/lib/Utils/IRRewrite.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Utils/IRRewrite.cpp -------------------------------------------------------------------------------- /compiler/lib/Utils/LoopUtils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Utils/LoopUtils.cpp -------------------------------------------------------------------------------- /compiler/lib/Utils/MemUtils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Utils/MemUtils.cpp -------------------------------------------------------------------------------- /compiler/lib/Utils/ModuleUtils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Utils/ModuleUtils.cpp -------------------------------------------------------------------------------- /compiler/lib/Utils/OpInterfaceUtils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Utils/OpInterfaceUtils.cpp -------------------------------------------------------------------------------- /compiler/lib/Utils/OptionUtils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Utils/OptionUtils.cpp -------------------------------------------------------------------------------- /compiler/lib/Utils/PatternMatch.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Utils/PatternMatch.cpp -------------------------------------------------------------------------------- /compiler/lib/Utils/PipelineUtils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Utils/PipelineUtils.cpp -------------------------------------------------------------------------------- /compiler/lib/Utils/TileUtils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Utils/TileUtils.cpp -------------------------------------------------------------------------------- /compiler/lib/Utils/TypeUtils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Utils/TypeUtils.cpp -------------------------------------------------------------------------------- /compiler/lib/Utils/Utils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/lib/Utils/Utils.cpp -------------------------------------------------------------------------------- /compiler/numerical/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/numerical/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/numerical/hlo/canonicalize_ext.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/numerical/hlo/canonicalize_ext.mlir -------------------------------------------------------------------------------- /compiler/numerical/hlo/conv_bn.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/numerical/hlo/conv_bn.mlir -------------------------------------------------------------------------------- /compiler/numerical/hlo/dot_bn.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/numerical/hlo/dot_bn.mlir -------------------------------------------------------------------------------- /compiler/numerical/hlo/hlo_fold.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/numerical/hlo/hlo_fold.mlir -------------------------------------------------------------------------------- /compiler/numerical/hlo/hlo_move_down.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/numerical/hlo/hlo_move_down.mlir -------------------------------------------------------------------------------- /compiler/numerical/hlo/hlo_simplify.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/numerical/hlo/hlo_simplify.mlir -------------------------------------------------------------------------------- /compiler/numerical/hlo/numerical_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/numerical/hlo/numerical_test.py -------------------------------------------------------------------------------- /compiler/numerical/lit.cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/numerical/lit.cfg.py -------------------------------------------------------------------------------- /compiler/numerical/lit.site.cfg.py.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/numerical/lit.site.cfg.py.in -------------------------------------------------------------------------------- /compiler/python/ByteIRModules.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/python/ByteIRModules.cpp -------------------------------------------------------------------------------- /compiler/python/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/python/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/python/byteir/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/python/byteir/README.md -------------------------------------------------------------------------------- /compiler/python/byteir/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/python/byteir/__init__.py -------------------------------------------------------------------------------- /compiler/python/byteir/_backend_registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/python/byteir/_backend_registry.py -------------------------------------------------------------------------------- /compiler/python/byteir/compile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/python/byteir/compile.py -------------------------------------------------------------------------------- /compiler/python/byteir/dialects/CatOps.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/python/byteir/dialects/CatOps.td -------------------------------------------------------------------------------- /compiler/python/byteir/dialects/cat/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/python/byteir/dialects/cat/__init__.py -------------------------------------------------------------------------------- /compiler/python/byteir/dialects/cat/ait_cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/python/byteir/dialects/cat/ait_cache.py -------------------------------------------------------------------------------- /compiler/python/byteir/dialects/cat/tit_cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/python/byteir/dialects/cat/tit_cache.py -------------------------------------------------------------------------------- /compiler/python/byteir/pattern_matches.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/python/byteir/pattern_matches.py -------------------------------------------------------------------------------- /compiler/python/byteir/tools/cat_executor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/python/byteir/tools/cat_executor.py -------------------------------------------------------------------------------- /compiler/python/byteir/tools/compiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/python/byteir/tools/compiler.py -------------------------------------------------------------------------------- /compiler/python/byteir/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/python/byteir/utils.py -------------------------------------------------------------------------------- /compiler/python/gen_version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/python/gen_version.py -------------------------------------------------------------------------------- /compiler/python/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/python/setup.py -------------------------------------------------------------------------------- /compiler/python/test/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/python/test/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/python/test/api/test_pattern_matches.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/python/test/api/test_pattern_matches.py -------------------------------------------------------------------------------- /compiler/python/test/api/test_py_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/python/test/api/test_py_api.py -------------------------------------------------------------------------------- /compiler/python/test/lit.cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/python/test/lit.cfg.py -------------------------------------------------------------------------------- /compiler/python/test/lit.site.cfg.py.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/python/test/lit.site.cfg.py.in -------------------------------------------------------------------------------- /compiler/python/version.txt: -------------------------------------------------------------------------------- 1 | 1.9.3.0 -------------------------------------------------------------------------------- /compiler/scripts/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/scripts/README.md -------------------------------------------------------------------------------- /compiler/scripts/gen_testcases.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/scripts/gen_testcases.py -------------------------------------------------------------------------------- /compiler/scripts/gen_testcases_and_check_diff.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/scripts/gen_testcases_and_check_diff.sh -------------------------------------------------------------------------------- /compiler/scripts/sync_to_runtime.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/scripts/sync_to_runtime.sh -------------------------------------------------------------------------------- /compiler/test/Analysis/testPrintLiveness.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Analysis/testPrintLiveness.mlir -------------------------------------------------------------------------------- /compiler/test/Analysis/testPrintUseRange.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Analysis/testPrintUseRange.mlir -------------------------------------------------------------------------------- /compiler/test/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/test/CPURunner/gelu.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/CPURunner/gelu.mlir -------------------------------------------------------------------------------- /compiler/test/CPURunner/repeatCustomCall.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/CPURunner/repeatCustomCall.mlir -------------------------------------------------------------------------------- /compiler/test/CPURunner/scatterTiling.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/CPURunner/scatterTiling.mlir -------------------------------------------------------------------------------- /compiler/test/Conversion/HloToCat/basic_ops.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Conversion/HloToCat/basic_ops.mlir -------------------------------------------------------------------------------- /compiler/test/Conversion/HloToCat/fused_ops.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Conversion/HloToCat/fused_ops.mlir -------------------------------------------------------------------------------- /compiler/test/Conversion/ToGPU/funcToGPU.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Conversion/ToGPU/funcToGPU.mlir -------------------------------------------------------------------------------- /compiler/test/Conversion/ToLinalg/fusionHlo.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Conversion/ToLinalg/fusionHlo.mlir -------------------------------------------------------------------------------- /compiler/test/Conversion/ToLinalg/reducef16.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Conversion/ToLinalg/reducef16.mlir -------------------------------------------------------------------------------- /compiler/test/Conversion/ToLinalg/simpleHlo.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Conversion/ToLinalg/simpleHlo.mlir -------------------------------------------------------------------------------- /compiler/test/Conversion/ToPTX/genPTXConfig.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Conversion/ToPTX/genPTXConfig.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/Ace/attrs.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/Ace/attrs.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/Ace/bufferize.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/Ace/bufferize.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/Ace/canonicalize.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/Ace/canonicalize.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/Ace/ops.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/Ace/ops.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/Affine/affineToMemRef.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/Affine/affineToMemRef.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/Byre/bert_transformer.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/Byre/bert_transformer.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/Byre/buffer_ops.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/Byre/buffer_ops.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/Byre/bufferize.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/Byre/bufferize.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/Byre/canonicalize.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/Byre/canonicalize.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/Byre/interface.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/Byre/interface.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/Byre/invalid.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/Byre/invalid.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/Cat/ops.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/Cat/ops.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/Ccl/ccl_bufferize.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/Ccl/ccl_bufferize.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/Ccl/ccl_canonicalize.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/Ccl/ccl_canonicalize.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/Ccl/ccl_move_down.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/Ccl/ccl_move_down.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/Ccl/invalid.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/Ccl/invalid.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/Ccl/ops.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/Ccl/ops.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/GPU/gpu-block-swizzle.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/GPU/gpu-block-swizzle.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/Lace/invalid.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/Lace/invalid.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/Lace/ops.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/Lace/ops.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/Linalg/annotate.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/Linalg/annotate.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/Linalg/bufferize.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/Linalg/bufferize.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/Linalg/extension.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/Linalg/extension.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/Linalg/fuse-attention.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/Linalg/fuse-attention.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/Linalg/generalization.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/Linalg/generalization.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/Linalg/opTiling1.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/Linalg/opTiling1.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/Linalg/opTiling2.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/Linalg/opTiling2.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/Linalg/prefetch.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/Linalg/prefetch.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/Linalg/transform-dev.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/Linalg/transform-dev.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/MemRef/canonicalize.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/MemRef/canonicalize.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/MemRef/layout.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/MemRef/layout.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/MemRef/removeCopy.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/MemRef/removeCopy.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/MemRef/simplifyView.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/MemRef/simplifyView.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/Mhlo/fusion.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/Mhlo/fusion.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/Mhlo/multi_return.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/Mhlo/multi_return.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/Mhlo/reduce.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/Mhlo/reduce.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/Mhlo/simple.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/Mhlo/simple.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/SCF/forallCollapsing.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/SCF/forallCollapsing.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/SCF/fuseNestedForall.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/SCF/fuseNestedForall.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/Shape/insertTieShape.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/Shape/insertTieShape.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/Transform/cleanup.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/Transform/cleanup.mlir -------------------------------------------------------------------------------- /compiler/test/Dialect/Transform/dump.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Dialect/Transform/dump.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/CUDA/AliasLikeGPU/input.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/CUDA/AliasLikeGPU/input.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/CUDA/AliasLikeGPU/template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/CUDA/AliasLikeGPU/template.py -------------------------------------------------------------------------------- /compiler/test/E2E/CUDA/BertTiny/BW/input.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/CUDA/BertTiny/BW/input.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/CUDA/BertTiny/FW/input.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/CUDA/BertTiny/FW/input.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/CUDA/CclInference/input.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/CUDA/CclInference/input.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/CUDA/MLPBasic/input.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/CUDA/MLPBasic/input.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/CUDA/MLPInference/input.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/CUDA/MLPInference/input.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/CUDA/MLPInference/template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/CUDA/MLPInference/template.py -------------------------------------------------------------------------------- /compiler/test/E2E/CUDA/NanoGPT/BW/input.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/CUDA/NanoGPT/BW/input.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/CUDA/NanoGPT/FW/input.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/CUDA/NanoGPT/FW/input.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/CUDA/ResNet18/BW/input.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/CUDA/ResNet18/BW/input.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/CUDA/ResNet18/BW/template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/CUDA/ResNet18/BW/template.py -------------------------------------------------------------------------------- /compiler/test/E2E/CUDA/ResNet18/FW/input.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/CUDA/ResNet18/FW/input.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/CUDA/ResNet18/FW/template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/CUDA/ResNet18/FW/template.py -------------------------------------------------------------------------------- /compiler/test/E2E/CUDA/ResNet18/Whole/input.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/CUDA/ResNet18/Whole/input.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/Host/AliasLike/00_Input.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/AliasLike/00_Input.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/Host/AliasLike/01_HostOpt.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/AliasLike/01_HostOpt.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/Host/AliasLike/02b_ToLLVM.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/AliasLike/02b_ToLLVM.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/Host/AliasLike/Output.ll: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/AliasLike/Output.ll -------------------------------------------------------------------------------- /compiler/test/E2E/Host/AliasLike/Output.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/AliasLike/Output.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/Host/AliasLike/template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/AliasLike/template.py -------------------------------------------------------------------------------- /compiler/test/E2E/Host/Case0/00_Input.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/Case0/00_Input.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/Host/Case0/01_HostOpt.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/Case0/01_HostOpt.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/Host/Case0/02a_ByreHost.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/Case0/02a_ByreHost.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/Host/Case0/02b_ToLLVM.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/Case0/02b_ToLLVM.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/Host/Case0/03b_ToLLVMIR.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/Case0/03b_ToLLVMIR.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/Host/Case0/Output.ll: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/Case0/Output.ll -------------------------------------------------------------------------------- /compiler/test/E2E/Host/Case0/Output.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/Case0/Output.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/Host/Case0/TotalPipeline.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/Case0/TotalPipeline.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/Host/Case0/template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/Case0/template.py -------------------------------------------------------------------------------- /compiler/test/E2E/Host/Case0_Bytecode/Output.bc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/Case0_Bytecode/Output.bc -------------------------------------------------------------------------------- /compiler/test/E2E/Host/Case1/00_Input.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/Case1/00_Input.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/Host/Case1/01_HostOpt.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/Case1/01_HostOpt.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/Host/Case1/02a_ByreHost.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/Case1/02a_ByreHost.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/Host/Case1/02b_ToLLVM.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/Case1/02b_ToLLVM.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/Host/Case1/03b_ToLLVMIR.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/Case1/03b_ToLLVMIR.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/Host/Case1/Output.ll: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/Case1/Output.ll -------------------------------------------------------------------------------- /compiler/test/E2E/Host/Case1/Output.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/Case1/Output.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/Host/Case1/TotalPipeline.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/Case1/TotalPipeline.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/Host/Case1/template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/Case1/template.py -------------------------------------------------------------------------------- /compiler/test/E2E/Host/RngNormal/00_Input.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/RngNormal/00_Input.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/Host/RngNormal/01_HostOpt.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/RngNormal/01_HostOpt.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/Host/RngNormal/02b_ToLLVM.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/RngNormal/02b_ToLLVM.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/Host/RngNormal/Output.ll: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/RngNormal/Output.ll -------------------------------------------------------------------------------- /compiler/test/E2E/Host/RngNormal/Output.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/RngNormal/Output.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/Host/RngNormal/template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/RngNormal/template.py -------------------------------------------------------------------------------- /compiler/test/E2E/Host/RngUniform/00_Input.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/RngUniform/00_Input.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/Host/RngUniform/Output.ll: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/RngUniform/Output.ll -------------------------------------------------------------------------------- /compiler/test/E2E/Host/RngUniform/Output.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/RngUniform/Output.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/Host/RngUniform/template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/RngUniform/template.py -------------------------------------------------------------------------------- /compiler/test/E2E/Host/Transpose/00_Input.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/Transpose/00_Input.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/Host/Transpose/01_HostOpt.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/Transpose/01_HostOpt.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/Host/Transpose/02b_ToLLVM.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/Transpose/02b_ToLLVM.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/Host/Transpose/Output.ll: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/Transpose/Output.ll -------------------------------------------------------------------------------- /compiler/test/E2E/Host/Transpose/Output.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/Transpose/Output.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/Host/Transpose/template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/Transpose/template.py -------------------------------------------------------------------------------- /compiler/test/E2E/Host/TypeCvt/00_Input.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/TypeCvt/00_Input.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/Host/TypeCvt/01_HostOpt.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/TypeCvt/01_HostOpt.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/Host/TypeCvt/02a_ByreHost.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/TypeCvt/02a_ByreHost.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/Host/TypeCvt/02b_ToLLVM.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/TypeCvt/02b_ToLLVM.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/Host/TypeCvt/03b_ToLLVMIR.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/TypeCvt/03b_ToLLVMIR.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/Host/TypeCvt/Output.ll: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/TypeCvt/Output.ll -------------------------------------------------------------------------------- /compiler/test/E2E/Host/TypeCvt/Output.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/TypeCvt/Output.mlir -------------------------------------------------------------------------------- /compiler/test/E2E/Host/TypeCvt/template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/E2E/Host/TypeCvt/template.py -------------------------------------------------------------------------------- /compiler/test/Ops/conv.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Ops/conv.mlir -------------------------------------------------------------------------------- /compiler/test/Ops/dot.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Ops/dot.mlir -------------------------------------------------------------------------------- /compiler/test/Pipelines/HloOpts/mlp.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Pipelines/HloOpts/mlp.mlir -------------------------------------------------------------------------------- /compiler/test/Pipelines/HloOpts/rng.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Pipelines/HloOpts/rng.mlir -------------------------------------------------------------------------------- /compiler/test/Pipelines/Host/ToLLVM/subview.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Pipelines/Host/ToLLVM/subview.mlir -------------------------------------------------------------------------------- /compiler/test/Pipelines/Host/ToLLVM/tanh.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Pipelines/Host/ToLLVM/tanh.mlir -------------------------------------------------------------------------------- /compiler/test/Stat/allocCnt.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Stat/allocCnt.mlir -------------------------------------------------------------------------------- /compiler/test/Stat/opCnt.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Stat/opCnt.mlir -------------------------------------------------------------------------------- /compiler/test/Stat/opTypes.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Stat/opTypes.mlir -------------------------------------------------------------------------------- /compiler/test/Target/CUDA/all.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Target/CUDA/all.mlir -------------------------------------------------------------------------------- /compiler/test/Target/CUDA/kernel.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Target/CUDA/kernel.mlir -------------------------------------------------------------------------------- /compiler/test/Target/Cpp/attrs.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Target/Cpp/attrs.mlir -------------------------------------------------------------------------------- /compiler/test/Target/Cpp/binary.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Target/Cpp/binary.mlir -------------------------------------------------------------------------------- /compiler/test/Target/Cpp/call.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Target/Cpp/call.mlir -------------------------------------------------------------------------------- /compiler/test/Target/Cpp/cast.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Target/Cpp/cast.mlir -------------------------------------------------------------------------------- /compiler/test/Target/Cpp/common-cpp.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Target/Cpp/common-cpp.mlir -------------------------------------------------------------------------------- /compiler/test/Target/Cpp/const.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Target/Cpp/const.mlir -------------------------------------------------------------------------------- /compiler/test/Target/Cpp/control_flow.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Target/Cpp/control_flow.mlir -------------------------------------------------------------------------------- /compiler/test/Target/Cpp/for.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Target/Cpp/for.mlir -------------------------------------------------------------------------------- /compiler/test/Target/Cpp/if.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Target/Cpp/if.mlir -------------------------------------------------------------------------------- /compiler/test/Target/Cpp/invalid.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Target/Cpp/invalid.mlir -------------------------------------------------------------------------------- /compiler/test/Target/Cpp/memref.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Target/Cpp/memref.mlir -------------------------------------------------------------------------------- /compiler/test/Target/Cpp/opaque_types.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Target/Cpp/opaque_types.mlir -------------------------------------------------------------------------------- /compiler/test/Target/Cpp/stdops.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Target/Cpp/stdops.mlir -------------------------------------------------------------------------------- /compiler/test/Target/Cpp/types.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Target/Cpp/types.mlir -------------------------------------------------------------------------------- /compiler/test/Target/PTX/fusionFuncToPTX.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Target/PTX/fusionFuncToPTX.mlir -------------------------------------------------------------------------------- /compiler/test/Transforms/cmae.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Transforms/cmae.mlir -------------------------------------------------------------------------------- /compiler/test/Transforms/collectFunc.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Transforms/collectFunc.mlir -------------------------------------------------------------------------------- /compiler/test/Transforms/funTag.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Transforms/funTag.mlir -------------------------------------------------------------------------------- /compiler/test/Transforms/graphCanonicalize.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Transforms/graphCanonicalize.mlir -------------------------------------------------------------------------------- /compiler/test/Transforms/insertUniqueId.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Transforms/insertUniqueId.mlir -------------------------------------------------------------------------------- /compiler/test/Transforms/loopTag.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Transforms/loopTag.mlir -------------------------------------------------------------------------------- /compiler/test/Transforms/loopUnrollUseDepth.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Transforms/loopUnrollUseDepth.mlir -------------------------------------------------------------------------------- /compiler/test/Transforms/memoryPlanning.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Transforms/memoryPlanning.mlir -------------------------------------------------------------------------------- /compiler/test/Transforms/oneShotBufferize.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Transforms/oneShotBufferize.mlir -------------------------------------------------------------------------------- /compiler/test/Transforms/removeFunTag.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Transforms/removeFunTag.mlir -------------------------------------------------------------------------------- /compiler/test/Transforms/removeFuncBody.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Transforms/removeFuncBody.mlir -------------------------------------------------------------------------------- /compiler/test/Transforms/rewriteOpToStdCall.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Transforms/rewriteOpToStdCall.mlir -------------------------------------------------------------------------------- /compiler/test/Transforms/setAllSpace.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Transforms/setAllSpace.mlir -------------------------------------------------------------------------------- /compiler/test/Transforms/setArgShape.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Transforms/setArgShape.mlir -------------------------------------------------------------------------------- /compiler/test/Transforms/setArgSpace.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Transforms/setArgSpace.mlir -------------------------------------------------------------------------------- /compiler/test/Transforms/setOpAndArgSpace.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Transforms/setOpAndArgSpace.mlir -------------------------------------------------------------------------------- /compiler/test/Transforms/setOpSpace.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Transforms/setOpSpace.mlir -------------------------------------------------------------------------------- /compiler/test/Transforms/shapeFuncOutlining.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Transforms/shapeFuncOutlining.mlir -------------------------------------------------------------------------------- /compiler/test/Transforms/shapeReification.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/Transforms/shapeReification.mlir -------------------------------------------------------------------------------- /compiler/test/lib/Analysis/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/lib/Analysis/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/test/lib/Analysis/TestPrintLiveness.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/lib/Analysis/TestPrintLiveness.cpp -------------------------------------------------------------------------------- /compiler/test/lib/Analysis/TestPrintUseRange.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/lib/Analysis/TestPrintUseRange.cpp -------------------------------------------------------------------------------- /compiler/test/lib/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/lib/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/test/lib/Interface/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/lib/Interface/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/test/lib/Transformation/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/lib/Transformation/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/test/lib/Utils/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/lib/Utils/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/test/lib/Utils/TestMergeTwoModules.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/lib/Utils/TestMergeTwoModules.cpp -------------------------------------------------------------------------------- /compiler/test/lit.cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/lit.cfg.py -------------------------------------------------------------------------------- /compiler/test/lit.site.cfg.py.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/test/lit.site.cfg.py.in -------------------------------------------------------------------------------- /compiler/tools/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/tools/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/tools/byteir-cpu-runner/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/tools/byteir-cpu-runner/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/tools/byteir-opt/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/tools/byteir-opt/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/tools/byteir-opt/byteir-opt.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/tools/byteir-opt/byteir-opt.cpp -------------------------------------------------------------------------------- /compiler/tools/byteir-stat/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/tools/byteir-stat/CMakeLists.txt -------------------------------------------------------------------------------- /compiler/tools/byteir-stat/byteir-stat.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/tools/byteir-stat/byteir-stat.cpp -------------------------------------------------------------------------------- /compiler/tools/byteir-translate/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/compiler/tools/byteir-translate/CMakeLists.txt -------------------------------------------------------------------------------- /docker/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/docker/Dockerfile -------------------------------------------------------------------------------- /external/TritonTemplate/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/external/TritonTemplate/.gitignore -------------------------------------------------------------------------------- /external/TritonTemplate/README.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /external/TritonTemplate/python/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/external/TritonTemplate/python/setup.py -------------------------------------------------------------------------------- /external/TritonTemplate/python/tritontemplate/_libinfo.py: -------------------------------------------------------------------------------- 1 | __version__ = "dev0" -------------------------------------------------------------------------------- /external/TritonTemplate/python/tritontemplate/backend/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /external/TritonTemplate/python/tritontemplate/backend/cuda/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /external/TritonTemplate/python/tritontemplate/backend/cuda/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /external/TritonTemplate/python/tritontemplate/testing/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /external/TritonTemplate/python/tritontemplate/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /external/half/LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/external/half/LICENSE.txt -------------------------------------------------------------------------------- /external/half/README.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/external/half/README.txt -------------------------------------------------------------------------------- /external/half/include/half/half.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/external/half/include/half/half.hpp -------------------------------------------------------------------------------- /external/patches/AITemplate/A10.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/external/patches/AITemplate/A10.patch -------------------------------------------------------------------------------- /external/patches/AITemplate/logging.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/external/patches/AITemplate/logging.patch -------------------------------------------------------------------------------- /external/patches/AITemplate/num_builders.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/external/patches/AITemplate/num_builders.patch -------------------------------------------------------------------------------- /external_libs/runtime/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/external_libs/runtime/CMakeLists.txt -------------------------------------------------------------------------------- /external_libs/runtime/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/external_libs/runtime/README.md -------------------------------------------------------------------------------- /external_libs/runtime/flash_attn/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(lib) -------------------------------------------------------------------------------- /external_libs/runtime/flash_attn/lib/alibi.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/external_libs/runtime/flash_attn/lib/alibi.h -------------------------------------------------------------------------------- /external_libs/runtime/flash_attn/lib/dropout.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/external_libs/runtime/flash_attn/lib/dropout.h -------------------------------------------------------------------------------- /external_libs/runtime/flash_attn/lib/flash.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/external_libs/runtime/flash_attn/lib/flash.h -------------------------------------------------------------------------------- /external_libs/runtime/flash_attn/lib/mask.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/external_libs/runtime/flash_attn/lib/mask.h -------------------------------------------------------------------------------- /external_libs/runtime/flash_attn/lib/philox.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/external_libs/runtime/flash_attn/lib/philox.cuh -------------------------------------------------------------------------------- /external_libs/runtime/flash_attn/lib/rotary.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/external_libs/runtime/flash_attn/lib/rotary.h -------------------------------------------------------------------------------- /external_libs/runtime/flash_attn/lib/softmax.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/external_libs/runtime/flash_attn/lib/softmax.h -------------------------------------------------------------------------------- /external_libs/runtime/flash_attn/lib/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/external_libs/runtime/flash_attn/lib/utils.h -------------------------------------------------------------------------------- /frontends/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/README.md -------------------------------------------------------------------------------- /frontends/onnx-frontend/.gitignore: -------------------------------------------------------------------------------- 1 | .pytest_cache/ 2 | *.pyc 3 | *.tar.gz 4 | 5 | build/ 6 | -------------------------------------------------------------------------------- /frontends/onnx-frontend/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/onnx-frontend/CMakeLists.txt -------------------------------------------------------------------------------- /frontends/onnx-frontend/MLIR.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/onnx-frontend/MLIR.cmake -------------------------------------------------------------------------------- /frontends/onnx-frontend/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/onnx-frontend/README.md -------------------------------------------------------------------------------- /frontends/onnx-frontend/onnx-frontend/src/Support/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_onnx_frontend_library(OFSupport 2 | OFUtils.cpp 3 | ) -------------------------------------------------------------------------------- /frontends/onnx-frontend/pytest.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/onnx-frontend/pytest.ini -------------------------------------------------------------------------------- /frontends/onnx-frontend/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/onnx-frontend/requirements.txt -------------------------------------------------------------------------------- /frontends/onnx-frontend/scripts/envsetup.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/onnx-frontend/scripts/envsetup.sh -------------------------------------------------------------------------------- /frontends/onnx-frontend/test/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /frontends/onnx-frontend/test/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/onnx-frontend/test/base.py -------------------------------------------------------------------------------- /frontends/onnx-frontend/test/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/onnx-frontend/test/env.py -------------------------------------------------------------------------------- /frontends/onnx-frontend/test/ops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /frontends/onnx-frontend/test/ops/test_math.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/onnx-frontend/test/ops/test_math.py -------------------------------------------------------------------------------- /frontends/onnx-frontend/test/ops/test_nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/onnx-frontend/test/ops/test_nn.py -------------------------------------------------------------------------------- /frontends/onnx-frontend/test/ops/test_rnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/onnx-frontend/test/ops/test_rnn.py -------------------------------------------------------------------------------- /frontends/onnx-frontend/test/ops/test_tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/onnx-frontend/test/ops/test_tensor.py -------------------------------------------------------------------------------- /frontends/onnx-frontend/test/ops/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/onnx-frontend/test/ops/utils.py -------------------------------------------------------------------------------- /frontends/tf-frontend/.bazelrc: -------------------------------------------------------------------------------- 1 | ./external/tensorflow/.bazelrc -------------------------------------------------------------------------------- /frontends/tf-frontend/.bazelversion: -------------------------------------------------------------------------------- 1 | ./external/tensorflow/.bazelversion -------------------------------------------------------------------------------- /frontends/tf-frontend/.gitignore: -------------------------------------------------------------------------------- 1 | /bazel-* 2 | example/.workspace/* 3 | -------------------------------------------------------------------------------- /frontends/tf-frontend/.tf_configure.bazelrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/tf-frontend/.tf_configure.bazelrc -------------------------------------------------------------------------------- /frontends/tf-frontend/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /frontends/tf-frontend/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/tf-frontend/README.md -------------------------------------------------------------------------------- /frontends/tf-frontend/WORKSPACE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/tf-frontend/WORKSPACE -------------------------------------------------------------------------------- /frontends/tf-frontend/byteir/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /frontends/tf-frontend/byteir/ace.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/tf-frontend/byteir/ace.BUILD -------------------------------------------------------------------------------- /frontends/tf-frontend/byteir/workspace.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/tf-frontend/byteir/workspace.bzl -------------------------------------------------------------------------------- /frontends/tf-frontend/docs/attributes.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/tf-frontend/docs/attributes.md -------------------------------------------------------------------------------- /frontends/tf-frontend/docs/developer_guild.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/tf-frontend/docs/developer_guild.md -------------------------------------------------------------------------------- /frontends/tf-frontend/example/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/tf-frontend/example/resnet.py -------------------------------------------------------------------------------- /frontends/tf-frontend/example/resnet50_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/tf-frontend/example/resnet50_model.py -------------------------------------------------------------------------------- /frontends/tf-frontend/scripts/apply_patches.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/tf-frontend/scripts/apply_patches.sh -------------------------------------------------------------------------------- /frontends/tf-frontend/scripts/build_and_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/tf-frontend/scripts/build_and_test.sh -------------------------------------------------------------------------------- /frontends/tf-frontend/scripts/prepare.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/tf-frontend/scripts/prepare.sh -------------------------------------------------------------------------------- /frontends/tf-frontend/tf_mlir_ext/tests/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/tf-frontend/tf_mlir_ext/tests/BUILD -------------------------------------------------------------------------------- /frontends/tf-frontend/tf_mlir_ext/utils/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/tf-frontend/tf_mlir_ext/utils/BUILD -------------------------------------------------------------------------------- /frontends/tf-frontend/tf_mlir_ext/utils/dce.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/tf-frontend/tf_mlir_ext/utils/dce.cc -------------------------------------------------------------------------------- /frontends/tf-frontend/tf_mlir_ext/utils/dce.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/tf-frontend/tf_mlir_ext/utils/dce.h -------------------------------------------------------------------------------- /frontends/tf-frontend/tf_mlir_ext/utils/utils.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/tf-frontend/tf_mlir_ext/utils/utils.cc -------------------------------------------------------------------------------- /frontends/tf-frontend/tf_mlir_ext/utils/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/tf-frontend/tf_mlir_ext/utils/utils.h -------------------------------------------------------------------------------- /frontends/tf-frontend/tools/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/tf-frontend/tools/BUILD -------------------------------------------------------------------------------- /frontends/tf-frontend/tools/tf_ext_opt_main.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/tf-frontend/tools/tf_ext_opt_main.cc -------------------------------------------------------------------------------- /frontends/tf-frontend/tools/tf_frontend_main.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/tf-frontend/tools/tf_frontend_main.cc -------------------------------------------------------------------------------- /frontends/tf-frontend/utils/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/tf-frontend/utils/BUILD -------------------------------------------------------------------------------- /frontends/tf-frontend/utils/attributes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/tf-frontend/utils/attributes.h -------------------------------------------------------------------------------- /frontends/tf-frontend/utils/graphdef_opt.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/tf-frontend/utils/graphdef_opt.cc -------------------------------------------------------------------------------- /frontends/tf-frontend/utils/graphdef_opt.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/tf-frontend/utils/graphdef_opt.h -------------------------------------------------------------------------------- /frontends/tf-frontend/utils/misc.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/tf-frontend/utils/misc.cc -------------------------------------------------------------------------------- /frontends/tf-frontend/utils/misc.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/tf-frontend/utils/misc.h -------------------------------------------------------------------------------- /frontends/torch-frontend/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/torch-frontend/.gitignore -------------------------------------------------------------------------------- /frontends/torch-frontend/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/torch-frontend/CMakeLists.txt -------------------------------------------------------------------------------- /frontends/torch-frontend/MLIR.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/torch-frontend/MLIR.cmake -------------------------------------------------------------------------------- /frontends/torch-frontend/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/torch-frontend/README.md -------------------------------------------------------------------------------- /frontends/torch-frontend/TorchMLIR.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/torch-frontend/TorchMLIR.cmake -------------------------------------------------------------------------------- /frontends/torch-frontend/build-requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/torch-frontend/build-requirements.txt -------------------------------------------------------------------------------- /frontends/torch-frontend/examples/demo/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/torch-frontend/examples/demo/README.md -------------------------------------------------------------------------------- /frontends/torch-frontend/examples/demo/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/torch-frontend/examples/demo/config.py -------------------------------------------------------------------------------- /frontends/torch-frontend/examples/demo/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/torch-frontend/examples/demo/main.py -------------------------------------------------------------------------------- /frontends/torch-frontend/scripts/build.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/torch-frontend/scripts/build.sh -------------------------------------------------------------------------------- /frontends/torch-frontend/scripts/envsetup.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/torch-frontend/scripts/envsetup.sh -------------------------------------------------------------------------------- /frontends/torch-frontend/test-requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/frontends/torch-frontend/test-requirements.txt -------------------------------------------------------------------------------- /frontends/torch-frontend/torch-frontend/python/version.txt: -------------------------------------------------------------------------------- 1 | 1.3.4 -------------------------------------------------------------------------------- /runtime/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/.gitignore -------------------------------------------------------------------------------- /runtime/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/README.md -------------------------------------------------------------------------------- /runtime/VERSION_NUMBER: -------------------------------------------------------------------------------- 1 | 1.9.3.0 2 | -------------------------------------------------------------------------------- /runtime/cmake/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/cmake/CMakeLists.txt -------------------------------------------------------------------------------- /runtime/cmake/Modules/FindNCCL.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/cmake/Modules/FindNCCL.cmake -------------------------------------------------------------------------------- /runtime/cmake/brt_common.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/cmake/brt_common.cmake -------------------------------------------------------------------------------- /runtime/cmake/brt_config.h.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/cmake/brt_config.h.in -------------------------------------------------------------------------------- /runtime/cmake/brt_device_cpu.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/cmake/brt_device_cpu.cmake -------------------------------------------------------------------------------- /runtime/cmake/brt_device_cuda.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/cmake/brt_device_cuda.cmake -------------------------------------------------------------------------------- /runtime/cmake/brt_device_nccl.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/cmake/brt_device_nccl.cmake -------------------------------------------------------------------------------- /runtime/cmake/brt_framework.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/cmake/brt_framework.cmake -------------------------------------------------------------------------------- /runtime/cmake/brt_ir.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/cmake/brt_ir.cmake -------------------------------------------------------------------------------- /runtime/cmake/brt_provider_cpu.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/cmake/brt_provider_cpu.cmake -------------------------------------------------------------------------------- /runtime/cmake/brt_provider_cuda.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/cmake/brt_provider_cuda.cmake -------------------------------------------------------------------------------- /runtime/cmake/brt_provider_nccl.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/cmake/brt_provider_nccl.cmake -------------------------------------------------------------------------------- /runtime/cmake/brt_python_bindings.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/cmake/brt_python_bindings.cmake -------------------------------------------------------------------------------- /runtime/cmake/brt_shared.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/cmake/brt_shared.cmake -------------------------------------------------------------------------------- /runtime/cmake/brt_unittests.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/cmake/brt_unittests.cmake -------------------------------------------------------------------------------- /runtime/examples/external_project/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/examples/external_project/CMakeLists.txt -------------------------------------------------------------------------------- /runtime/examples/external_project/main.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/examples/external_project/main.cpp -------------------------------------------------------------------------------- /runtime/include/brt/backends/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/include/brt/backends/README.md -------------------------------------------------------------------------------- /runtime/include/brt/backends/common.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/include/brt/backends/common.h -------------------------------------------------------------------------------- /runtime/include/brt/backends/nccl/device/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/include/brt/backends/nccl/device/utils.h -------------------------------------------------------------------------------- /runtime/include/brt/backends/rng_state_context.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/include/brt/backends/rng_state_context.h -------------------------------------------------------------------------------- /runtime/include/brt/core/common/code_location.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/include/brt/core/common/code_location.h -------------------------------------------------------------------------------- /runtime/include/brt/core/common/common.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/include/brt/core/common/common.h -------------------------------------------------------------------------------- /runtime/include/brt/core/common/enums.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/include/brt/core/common/enums.h -------------------------------------------------------------------------------- /runtime/include/brt/core/common/exceptions.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/include/brt/core/common/exceptions.h -------------------------------------------------------------------------------- /runtime/include/brt/core/common/logging/isink.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/include/brt/core/common/logging/isink.h -------------------------------------------------------------------------------- /runtime/include/brt/core/common/logging/macros.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/include/brt/core/common/logging/macros.h -------------------------------------------------------------------------------- /runtime/include/brt/core/common/make_string.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/include/brt/core/common/make_string.h -------------------------------------------------------------------------------- /runtime/include/brt/core/common/status.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/include/brt/core/common/status.h -------------------------------------------------------------------------------- /runtime/include/brt/core/common/string_view.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/include/brt/core/common/string_view.h -------------------------------------------------------------------------------- /runtime/include/brt/core/context/work_queue.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/include/brt/core/context/work_queue.h -------------------------------------------------------------------------------- /runtime/include/brt/core/distributed/d_context.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/include/brt/core/distributed/d_context.h -------------------------------------------------------------------------------- /runtime/include/brt/core/framework/allocator.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/include/brt/core/framework/allocator.h -------------------------------------------------------------------------------- /runtime/include/brt/core/framework/arena.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/include/brt/core/framework/arena.h -------------------------------------------------------------------------------- /runtime/include/brt/core/framework/bfc_arena.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/include/brt/core/framework/bfc_arena.h -------------------------------------------------------------------------------- /runtime/include/brt/core/framework/brt_mutex.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/include/brt/core/framework/brt_mutex.h -------------------------------------------------------------------------------- /runtime/include/brt/core/framework/device_api.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/include/brt/core/framework/device_api.h -------------------------------------------------------------------------------- /runtime/include/brt/core/framework/dtype.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/include/brt/core/framework/dtype.h -------------------------------------------------------------------------------- /runtime/include/brt/core/framework/event.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/include/brt/core/framework/event.h -------------------------------------------------------------------------------- /runtime/include/brt/core/framework/memory_info.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/include/brt/core/framework/memory_info.h -------------------------------------------------------------------------------- /runtime/include/brt/core/framework/op_accessor.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/include/brt/core/framework/op_accessor.h -------------------------------------------------------------------------------- /runtime/include/brt/core/framework/op_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/include/brt/core/framework/op_kernel.h -------------------------------------------------------------------------------- /runtime/include/brt/core/framework/value.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/include/brt/core/framework/value.h -------------------------------------------------------------------------------- /runtime/include/brt/core/ir/builder.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/include/brt/core/ir/builder.h -------------------------------------------------------------------------------- /runtime/include/brt/core/ir/engine_util.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/include/brt/core/ir/engine_util.h -------------------------------------------------------------------------------- /runtime/include/brt/core/ir/graph_info.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/include/brt/core/ir/graph_info.h -------------------------------------------------------------------------------- /runtime/include/brt/core/ir/ir.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/include/brt/core/ir/ir.h -------------------------------------------------------------------------------- /runtime/include/brt/core/ir/op_helper.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/include/brt/core/ir/op_helper.h -------------------------------------------------------------------------------- /runtime/include/brt/core/ir/util.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/include/brt/core/ir/util.h -------------------------------------------------------------------------------- /runtime/include/brt/core/session/session.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/include/brt/core/session/session.h -------------------------------------------------------------------------------- /runtime/lib/backends/cpu/device/llvm/jit.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/lib/backends/cpu/device/llvm/jit.cc -------------------------------------------------------------------------------- /runtime/lib/backends/cuda/device/common/util.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/lib/backends/cuda/device/common/util.cc -------------------------------------------------------------------------------- /runtime/lib/backends/cuda/device/compile/ptx.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/lib/backends/cuda/device/compile/ptx.cc -------------------------------------------------------------------------------- /runtime/lib/backends/cuda/device/cuda_env.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/lib/backends/cuda/device/cuda_env.cc -------------------------------------------------------------------------------- /runtime/lib/backends/nccl/device/utils.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/lib/backends/nccl/device/utils.cc -------------------------------------------------------------------------------- /runtime/lib/backends/nccl/providers/all_gather.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/lib/backends/nccl/providers/all_gather.h -------------------------------------------------------------------------------- /runtime/lib/backends/nccl/providers/all_reduce.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/lib/backends/nccl/providers/all_reduce.h -------------------------------------------------------------------------------- /runtime/lib/backends/nccl/providers/broadcast.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/lib/backends/nccl/providers/broadcast.cc -------------------------------------------------------------------------------- /runtime/lib/backends/nccl/providers/broadcast.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/lib/backends/nccl/providers/broadcast.h -------------------------------------------------------------------------------- /runtime/lib/backends/nccl/providers/recv.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/lib/backends/nccl/providers/recv.cc -------------------------------------------------------------------------------- /runtime/lib/backends/nccl/providers/recv.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/lib/backends/nccl/providers/recv.h -------------------------------------------------------------------------------- /runtime/lib/backends/nccl/providers/send.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/lib/backends/nccl/providers/send.cc -------------------------------------------------------------------------------- /runtime/lib/backends/nccl/providers/send.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/lib/backends/nccl/providers/send.h -------------------------------------------------------------------------------- /runtime/lib/core/common/common.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/lib/core/common/common.cc -------------------------------------------------------------------------------- /runtime/lib/core/common/logging/capture.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/lib/core/common/logging/capture.cc -------------------------------------------------------------------------------- /runtime/lib/core/common/logging/logging.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/lib/core/common/logging/logging.cc -------------------------------------------------------------------------------- /runtime/lib/core/common/status.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/lib/core/common/status.cc -------------------------------------------------------------------------------- /runtime/lib/core/common/utils/math_helper.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/lib/core/common/utils/math_helper.cc -------------------------------------------------------------------------------- /runtime/lib/core/context/execution_frame.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/lib/core/context/execution_frame.cc -------------------------------------------------------------------------------- /runtime/lib/core/framework/allocator.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/lib/core/framework/allocator.cc -------------------------------------------------------------------------------- /runtime/lib/core/framework/bfc_arena.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/lib/core/framework/bfc_arena.cc -------------------------------------------------------------------------------- /runtime/lib/core/framework/device_api.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/lib/core/framework/device_api.cc -------------------------------------------------------------------------------- /runtime/lib/core/framework/execution_plan.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/lib/core/framework/execution_plan.cc -------------------------------------------------------------------------------- /runtime/lib/core/framework/execution_provider.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/lib/core/framework/execution_provider.cc -------------------------------------------------------------------------------- /runtime/lib/core/framework/kernel_registry.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/lib/core/framework/kernel_registry.cc -------------------------------------------------------------------------------- /runtime/lib/core/framework/op_accessor.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/lib/core/framework/op_accessor.cc -------------------------------------------------------------------------------- /runtime/lib/core/framework/op_kernel_info.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/lib/core/framework/op_kernel_info.cc -------------------------------------------------------------------------------- /runtime/lib/core/ir/builder.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/lib/core/ir/builder.cc -------------------------------------------------------------------------------- /runtime/lib/core/ir/ir.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/lib/core/ir/ir.cc -------------------------------------------------------------------------------- /runtime/lib/core/ir/op_helper.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/lib/core/ir/op_helper.cc -------------------------------------------------------------------------------- /runtime/lib/core/ir/util.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/lib/core/ir/util.cc -------------------------------------------------------------------------------- /runtime/lib/core/session/request_context.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/lib/core/session/request_context.cc -------------------------------------------------------------------------------- /runtime/lib/core/session/session.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/lib/core/session/session.cc -------------------------------------------------------------------------------- /runtime/python/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/python/README.md -------------------------------------------------------------------------------- /runtime/python/brt/__init__.py: -------------------------------------------------------------------------------- 1 | from ._brt import * 2 | -------------------------------------------------------------------------------- /runtime/python/brt/backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/python/brt/backend.py -------------------------------------------------------------------------------- /runtime/python/brt/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/python/brt/utils.py -------------------------------------------------------------------------------- /runtime/python/examples/add2.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/python/examples/add2.mlir -------------------------------------------------------------------------------- /runtime/python/examples/add2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/python/examples/add2.py -------------------------------------------------------------------------------- /runtime/python/examples/ait_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/python/examples/ait_op.py -------------------------------------------------------------------------------- /runtime/python/examples/arg_alias.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/python/examples/arg_alias.mlir -------------------------------------------------------------------------------- /runtime/python/examples/arg_alias.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/python/examples/arg_alias.py -------------------------------------------------------------------------------- /runtime/python/examples/distribute_mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/python/examples/distribute_mlp.py -------------------------------------------------------------------------------- /runtime/python/examples/llm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/python/examples/llm.py -------------------------------------------------------------------------------- /runtime/python/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/python/setup.py -------------------------------------------------------------------------------- /runtime/python/src/module.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/python/src/module.cc -------------------------------------------------------------------------------- /runtime/test/backends/cuda/device/nvrtc_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/backends/cuda/device/nvrtc_test.cc -------------------------------------------------------------------------------- /runtime/test/backends/cuda/device/ptx_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/backends/cuda/device/ptx_test.cc -------------------------------------------------------------------------------- /runtime/test/backends/cuda/device/test_kernels.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/backends/cuda/device/test_kernels.h -------------------------------------------------------------------------------- /runtime/test/backends/nccl/device/test_utils.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/backends/nccl/device/test_utils.cc -------------------------------------------------------------------------------- /runtime/test/common/env.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/common/env.cc -------------------------------------------------------------------------------- /runtime/test/common/models.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/common/models.cc -------------------------------------------------------------------------------- /runtime/test/common/util.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/common/util.cc -------------------------------------------------------------------------------- /runtime/test/context/exec_frame_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/context/exec_frame_test.cc -------------------------------------------------------------------------------- /runtime/test/exported.ld: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/exported.ld -------------------------------------------------------------------------------- /runtime/test/external_kernels/cpu/kernels.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/external_kernels/cpu/kernels.cc -------------------------------------------------------------------------------- /runtime/test/external_kernels/cuda/kernels.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/external_kernels/cuda/kernels.cc -------------------------------------------------------------------------------- /runtime/test/external_kernels/cuda/kernels.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/external_kernels/cuda/kernels.cu -------------------------------------------------------------------------------- /runtime/test/external_kernels/cuda/kernels.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/external_kernels/cuda/kernels.h -------------------------------------------------------------------------------- /runtime/test/framework/allocator_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/framework/allocator_test.cc -------------------------------------------------------------------------------- /runtime/test/framework/misc.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/framework/misc.cc -------------------------------------------------------------------------------- /runtime/test/include/brt/test/common/config.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/include/brt/test/common/config.h -------------------------------------------------------------------------------- /runtime/test/include/brt/test/common/cuda/util.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/include/brt/test/common/cuda/util.h -------------------------------------------------------------------------------- /runtime/test/include/brt/test/common/env.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/include/brt/test/common/env.h -------------------------------------------------------------------------------- /runtime/test/include/brt/test/common/models.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/include/brt/test/common/models.h -------------------------------------------------------------------------------- /runtime/test/include/brt/test/common/util.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/include/brt/test/common/util.h -------------------------------------------------------------------------------- /runtime/test/ir/builder_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/ir/builder_test.cc -------------------------------------------------------------------------------- /runtime/test/ir/ir_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/ir/ir_test.cc -------------------------------------------------------------------------------- /runtime/test/session/session_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/session/session_test.cc -------------------------------------------------------------------------------- /runtime/test/test_files/AITOp/permute_a100.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/test_files/AITOp/permute_a100.so -------------------------------------------------------------------------------- /runtime/test/test_files/AITOp/permute_entry.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/test_files/AITOp/permute_entry.mlir -------------------------------------------------------------------------------- /runtime/test/test_files/Distributed/ccl.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/test_files/Distributed/ccl.mlir -------------------------------------------------------------------------------- /runtime/test/test_files/Distributed/ccl.ptx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/test_files/Distributed/ccl.ptx -------------------------------------------------------------------------------- /runtime/test/test_files/Distributed/recv.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/test_files/Distributed/recv.mlir -------------------------------------------------------------------------------- /runtime/test/test_files/Distributed/send.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/test_files/Distributed/send.mlir -------------------------------------------------------------------------------- /runtime/test/test_files/LLJIT/Case0/entry.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/test_files/LLJIT/Case0/entry.mlir -------------------------------------------------------------------------------- /runtime/test/test_files/LLJIT/add.ll: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/test_files/LLJIT/add.ll -------------------------------------------------------------------------------- /runtime/test/test_files/LLJIT/tanh.ll: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/test_files/LLJIT/tanh.ll -------------------------------------------------------------------------------- /runtime/test/test_files/LLJIT/typecvt.ll: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/test_files/LLJIT/typecvt.ll -------------------------------------------------------------------------------- /runtime/test/test_files/add2_cpu.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/test_files/add2_cpu.mlir -------------------------------------------------------------------------------- /runtime/test/test_files/cuda_add.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/test_files/cuda_add.cu -------------------------------------------------------------------------------- /runtime/test/test_files/custom_add_cpu2cuda.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/test_files/custom_add_cpu2cuda.mlir -------------------------------------------------------------------------------- /runtime/test/test_files/fill_cuda.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/test_files/fill_cuda.mlir -------------------------------------------------------------------------------- /runtime/test/test_files/flash_attn_bwd.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/test_files/flash_attn_bwd.mlir -------------------------------------------------------------------------------- /runtime/test/test_files/flash_attn_fwd.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/test_files/flash_attn_fwd.mlir -------------------------------------------------------------------------------- /runtime/test/test_files/flash_attn_inputs_k.data: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/test_files/flash_attn_inputs_k.data -------------------------------------------------------------------------------- /runtime/test/test_files/flash_attn_inputs_q.data: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/test_files/flash_attn_inputs_q.data -------------------------------------------------------------------------------- /runtime/test/test_files/flash_attn_inputs_v.data: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/test_files/flash_attn_inputs_v.data -------------------------------------------------------------------------------- /runtime/test/test_files/flash_attn_kvcache.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/test_files/flash_attn_kvcache.mlir -------------------------------------------------------------------------------- /runtime/test/test_files/flash_attn_kvcache_inputs_cache_seqlens.data: -------------------------------------------------------------------------------- 1 | 64 64 -------------------------------------------------------------------------------- /runtime/test/test_files/llvm_ptx_add.ptx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/test_files/llvm_ptx_add.ptx -------------------------------------------------------------------------------- /runtime/test/test_files/nvcc_ptx_add.ptx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/test_files/nvcc_ptx_add.ptx -------------------------------------------------------------------------------- /runtime/test/test_files/resnet18_bw_device.ptx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/test_files/resnet18_bw_device.ptx -------------------------------------------------------------------------------- /runtime/test/test_files/resnet18_fw_device.ptx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/test_files/resnet18_fw_device.ptx -------------------------------------------------------------------------------- /runtime/test/test_files/rng_cuda.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/test_files/rng_cuda.mlir -------------------------------------------------------------------------------- /runtime/test/test_files/rng_state_cpu.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/test_files/rng_state_cpu.mlir -------------------------------------------------------------------------------- /runtime/test/test_files/rng_state_cuda.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/test_files/rng_state_cuda.mlir -------------------------------------------------------------------------------- /runtime/test/test_files/string_equal.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/test_files/string_equal.mlir -------------------------------------------------------------------------------- /runtime/test/test_files/string_equal_scalar.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/test_files/string_equal_scalar.mlir -------------------------------------------------------------------------------- /runtime/test/unittest_main/test_main.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/test/unittest_main/test_main.cc -------------------------------------------------------------------------------- /runtime/version.ld: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/runtime/version.ld -------------------------------------------------------------------------------- /scripts/apply_patches.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/scripts/apply_patches.sh -------------------------------------------------------------------------------- /scripts/clang_format_check.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/scripts/clang_format_check.sh -------------------------------------------------------------------------------- /scripts/compiler/build_and_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/scripts/compiler/build_and_test.sh -------------------------------------------------------------------------------- /scripts/format_check.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/scripts/format_check.py -------------------------------------------------------------------------------- /scripts/prepare.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/scripts/prepare.sh -------------------------------------------------------------------------------- /scripts/runtime/build_and_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/scripts/runtime/build_and_test.sh -------------------------------------------------------------------------------- /scripts/runtime/build_external_project.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/scripts/runtime/build_external_project.sh -------------------------------------------------------------------------------- /talks/ChinaSoftCon-ByteIR.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/talks/ChinaSoftCon-ByteIR.pdf -------------------------------------------------------------------------------- /talks/c4ml23_poster.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/talks/c4ml23_poster.pdf -------------------------------------------------------------------------------- /tests/build_and_test_e2e.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/tests/build_and_test_e2e.sh -------------------------------------------------------------------------------- /tests/compatibility_test/execute.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/tests/compatibility_test/execute.py -------------------------------------------------------------------------------- /tests/compatibility_test/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/tests/compatibility_test/main.py -------------------------------------------------------------------------------- /tests/compatibility_test/reporting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/tests/compatibility_test/reporting.py -------------------------------------------------------------------------------- /tests/numerical_test/execute.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/tests/numerical_test/execute.py -------------------------------------------------------------------------------- /tests/numerical_test/gen_brt_tests.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/tests/numerical_test/gen_brt_tests.py -------------------------------------------------------------------------------- /tests/numerical_test/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/tests/numerical_test/main.py -------------------------------------------------------------------------------- /tests/numerical_test/mlir_tests/cpu_ops/add.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/tests/numerical_test/mlir_tests/cpu_ops/add.mlir -------------------------------------------------------------------------------- /tests/numerical_test/mlir_tests/cpu_ops/rng.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/tests/numerical_test/mlir_tests/cpu_ops/rng.mlir -------------------------------------------------------------------------------- /tests/numerical_test/mlir_tests/ops/add.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/tests/numerical_test/mlir_tests/ops/add.mlir -------------------------------------------------------------------------------- /tests/numerical_test/mlir_tests/ops/bmm_rcr.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/tests/numerical_test/mlir_tests/ops/bmm_rcr.mlir -------------------------------------------------------------------------------- /tests/numerical_test/mlir_tests/ops/bmm_rrc.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/tests/numerical_test/mlir_tests/ops/bmm_rrc.mlir -------------------------------------------------------------------------------- /tests/numerical_test/mlir_tests/ops/concat.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/tests/numerical_test/mlir_tests/ops/concat.mlir -------------------------------------------------------------------------------- /tests/numerical_test/mlir_tests/ops/concat2.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/tests/numerical_test/mlir_tests/ops/concat2.mlir -------------------------------------------------------------------------------- /tests/numerical_test/mlir_tests/ops/divide.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/tests/numerical_test/mlir_tests/ops/divide.mlir -------------------------------------------------------------------------------- /tests/numerical_test/mlir_tests/ops/gather.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/tests/numerical_test/mlir_tests/ops/gather.mlir -------------------------------------------------------------------------------- /tests/numerical_test/mlir_tests/ops/mul_f16.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/tests/numerical_test/mlir_tests/ops/mul_f16.mlir -------------------------------------------------------------------------------- /tests/numerical_test/mlir_tests/ops/mul_f32.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/tests/numerical_test/mlir_tests/ops/mul_f32.mlir -------------------------------------------------------------------------------- /tests/numerical_test/mlir_tests/ops/negate.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/tests/numerical_test/mlir_tests/ops/negate.mlir -------------------------------------------------------------------------------- /tests/numerical_test/mlir_tests/ops/power.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/tests/numerical_test/mlir_tests/ops/power.mlir -------------------------------------------------------------------------------- /tests/numerical_test/mlir_tests/ops/rsqrt.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/tests/numerical_test/mlir_tests/ops/rsqrt.mlir -------------------------------------------------------------------------------- /tests/numerical_test/mlir_tests/ops/scatter.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/tests/numerical_test/mlir_tests/ops/scatter.mlir -------------------------------------------------------------------------------- /tests/numerical_test/mlir_tests/ops/select.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/tests/numerical_test/mlir_tests/ops/select.mlir -------------------------------------------------------------------------------- /tests/numerical_test/mlir_tests/ops/slice.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/tests/numerical_test/mlir_tests/ops/slice.mlir -------------------------------------------------------------------------------- /tests/numerical_test/mlir_tests/ops/softmax.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/tests/numerical_test/mlir_tests/ops/softmax.mlir -------------------------------------------------------------------------------- /tests/numerical_test/profiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/tests/numerical_test/profiler.py -------------------------------------------------------------------------------- /tests/numerical_test/reporting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/tests/numerical_test/reporting.py -------------------------------------------------------------------------------- /tests/numerical_test/testset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/byteir/HEAD/tests/numerical_test/testset.py --------------------------------------------------------------------------------