├── .bazelignore ├── .bazelrc ├── .clang-format ├── .clang-format-ignore ├── .github ├── actions │ └── setup-build │ │ └── action.yml └── workflows │ ├── bazelBuildAndTestLlvm.yml │ ├── bazelBuildAndTestStablehlo.yml │ ├── bazelBuildAndTestTcp.yml │ └── bazelBuildAndTestTorchmlir.yml ├── .gitignore ├── .vscode └── settings.json ├── BUILD ├── LICENSE ├── README.md ├── WORKSPACE ├── deps.bzl ├── docker ├── Dockerfile └── run_docker.sh ├── docs └── gather.md ├── include └── mlir-tcp │ ├── Conversion │ ├── Passes.h │ ├── Passes.td │ ├── StablehloToTcp │ │ └── StablehloToTcp.h │ ├── TcpToArith │ │ └── TcpToArith.h │ ├── TcpToLinalg │ │ └── TcpToLinalg.h │ ├── TcpToTensor │ │ └── TcpToTensor.h │ └── TorchToTcp │ │ ├── TorchToTcp.h │ │ └── TorchToTcpCustomOp.h │ ├── Dialect │ ├── IR │ │ ├── TcpBase.td │ │ ├── TcpDialect.h │ │ ├── TcpEnums.td │ │ ├── TcpOps.h │ │ ├── TcpOps.td │ │ ├── TcpTilingInterfaceImpl.h │ │ └── TcpTypes.td │ └── Transforms │ │ ├── DropSymbolicShapeOpsPass.h │ │ ├── FuseTcpOpsPass.h │ │ ├── FusionPatterns.h │ │ ├── IsolateGroupOpsPass.h │ │ ├── Passes.h │ │ ├── Passes.td │ │ ├── TransformTensorOps.h │ │ └── VerifyTcpBackendContractPass.h │ ├── InitAll.h │ └── Pipeline │ └── Pipeline.h ├── lib ├── Conversion │ ├── PassDetail.h │ ├── Passes.cpp │ ├── StablehloToTcp │ │ └── StablehloToTcp.cpp │ ├── TcpToArith │ │ └── TcpToArith.cpp │ ├── TcpToLinalg │ │ ├── DataMovement.cpp │ │ ├── Elementwise.cpp │ │ ├── Misc.cpp │ │ ├── PopulatePatterns.h │ │ └── TcpToLinalg.cpp │ ├── TcpToTensor │ │ └── TcpToTensor.cpp │ └── TorchToTcp │ │ ├── DataMovement.cpp │ │ ├── Elementwise.cpp │ │ ├── Misc.cpp │ │ ├── PopulatePatterns.h │ │ ├── TcpCustomOp.cpp │ │ ├── TorchToTcp.cpp │ │ ├── TorchToTcpCustomOp.cpp │ │ ├── Utils.cpp │ │ └── Utils.h ├── Dialect │ ├── IR │ │ ├── TcpDialect.cpp │ │ ├── TcpOps.cpp │ │ └── TcpTilingInterfaceImpl.cpp │ └── Transforms │ │ ├── DropSymbolicShapeOpsPass.cpp │ │ ├── FuseTcpOpsPass.cpp │ │ ├── FusionPatterns.cpp │ │ ├── IsolateGroupOpsPass.cpp │ │ ├── PassDetail.h │ │ ├── Passes.cpp │ │ ├── TransformTensorOps.cpp │ │ └── VerifyTcpBackendContractPass.cpp ├── InitAll.cpp └── Pipeline │ └── Pipeline.cpp ├── local_repos.bzl ├── requirements.txt ├── requirements_lock.txt ├── test ├── AotCompile │ ├── BUILD │ ├── basic_tcp_ops.mlir │ ├── basic_tcp_ops_compile_execute_test.cpp │ └── model_loader_lib.py ├── BUILD ├── Conversion │ ├── StablehloToTcp │ │ └── stablehlo.mlir │ ├── TcpToArith │ │ └── basic.mlir │ ├── TcpToLinalg │ │ ├── binary.mlir │ │ ├── data_movement.mlir │ │ ├── misc.mlir │ │ └── unary.mlir │ ├── TcpToTensor │ │ └── data_movement.mlir │ └── TorchToTcp │ │ ├── data_movement.mlir │ │ ├── dynamo_fx_import.mlir │ │ ├── elementwise.mlir │ │ ├── misc.mlir │ │ ├── partial_conversion.mlir │ │ └── tcp_custom_ops.mlir ├── Dialect │ ├── binary.mlir │ ├── canonicalize.mlir │ ├── data_movement.mlir │ ├── decompose_tensor_ops.mlir │ ├── drop_symbolic_shape_ops.mlir │ ├── misc.mlir │ ├── structured_fuse.mlir │ ├── tcp_custom_ops.mlir │ ├── tcp_fusion.mlir │ ├── tcp_isolate_groups.mlir │ └── unary.mlir ├── Pipeline │ ├── tcp_to_llvm_pipeline.mlir │ └── torch_to_tcp_pipeline.mlir ├── lit.cfg.py ├── lit.site.cfg.py.in ├── python │ ├── BUILD │ └── fx_import │ │ └── basic_test.py └── python_lit │ └── fx_import │ ├── basic_test.py │ ├── custom_op_test.py │ └── symbolic_shape_expr_test.py ├── third_party ├── .clang-format ├── BUILD └── cnpy.BUILD └── tools ├── aot ├── BUILD ├── README.md ├── abi.h ├── aot_compile.bzl ├── execute_test.template.cpp ├── execute_test_generator.py ├── torch_exporter_harness.py └── torch_loader_utils.py ├── buildifier └── BUILD ├── clangd └── BUILD ├── pip └── BUILD └── tcp-opt └── tcp-opt.cpp /.bazelignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/.bazelignore -------------------------------------------------------------------------------- /.bazelrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/.bazelrc -------------------------------------------------------------------------------- /.clang-format: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/.clang-format -------------------------------------------------------------------------------- /.clang-format-ignore: -------------------------------------------------------------------------------- 1 | tools/aot/execute_test.template.cpp 2 | -------------------------------------------------------------------------------- /.github/actions/setup-build/action.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/.github/actions/setup-build/action.yml -------------------------------------------------------------------------------- /.github/workflows/bazelBuildAndTestLlvm.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/.github/workflows/bazelBuildAndTestLlvm.yml -------------------------------------------------------------------------------- /.github/workflows/bazelBuildAndTestStablehlo.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/.github/workflows/bazelBuildAndTestStablehlo.yml -------------------------------------------------------------------------------- /.github/workflows/bazelBuildAndTestTcp.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/.github/workflows/bazelBuildAndTestTcp.yml -------------------------------------------------------------------------------- /.github/workflows/bazelBuildAndTestTorchmlir.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/.github/workflows/bazelBuildAndTestTorchmlir.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/.gitignore -------------------------------------------------------------------------------- /.vscode/settings.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/.vscode/settings.json -------------------------------------------------------------------------------- /BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/BUILD -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/README.md -------------------------------------------------------------------------------- /WORKSPACE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/WORKSPACE -------------------------------------------------------------------------------- /deps.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/deps.bzl -------------------------------------------------------------------------------- /docker/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/docker/Dockerfile -------------------------------------------------------------------------------- /docker/run_docker.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/docker/run_docker.sh -------------------------------------------------------------------------------- /docs/gather.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/docs/gather.md -------------------------------------------------------------------------------- /include/mlir-tcp/Conversion/Passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/include/mlir-tcp/Conversion/Passes.h -------------------------------------------------------------------------------- /include/mlir-tcp/Conversion/Passes.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/include/mlir-tcp/Conversion/Passes.td -------------------------------------------------------------------------------- /include/mlir-tcp/Conversion/StablehloToTcp/StablehloToTcp.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/include/mlir-tcp/Conversion/StablehloToTcp/StablehloToTcp.h -------------------------------------------------------------------------------- /include/mlir-tcp/Conversion/TcpToArith/TcpToArith.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/include/mlir-tcp/Conversion/TcpToArith/TcpToArith.h -------------------------------------------------------------------------------- /include/mlir-tcp/Conversion/TcpToLinalg/TcpToLinalg.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/include/mlir-tcp/Conversion/TcpToLinalg/TcpToLinalg.h -------------------------------------------------------------------------------- /include/mlir-tcp/Conversion/TcpToTensor/TcpToTensor.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/include/mlir-tcp/Conversion/TcpToTensor/TcpToTensor.h -------------------------------------------------------------------------------- /include/mlir-tcp/Conversion/TorchToTcp/TorchToTcp.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/include/mlir-tcp/Conversion/TorchToTcp/TorchToTcp.h -------------------------------------------------------------------------------- /include/mlir-tcp/Conversion/TorchToTcp/TorchToTcpCustomOp.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/include/mlir-tcp/Conversion/TorchToTcp/TorchToTcpCustomOp.h -------------------------------------------------------------------------------- /include/mlir-tcp/Dialect/IR/TcpBase.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/include/mlir-tcp/Dialect/IR/TcpBase.td -------------------------------------------------------------------------------- /include/mlir-tcp/Dialect/IR/TcpDialect.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/include/mlir-tcp/Dialect/IR/TcpDialect.h -------------------------------------------------------------------------------- /include/mlir-tcp/Dialect/IR/TcpEnums.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/include/mlir-tcp/Dialect/IR/TcpEnums.td -------------------------------------------------------------------------------- /include/mlir-tcp/Dialect/IR/TcpOps.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/include/mlir-tcp/Dialect/IR/TcpOps.h -------------------------------------------------------------------------------- /include/mlir-tcp/Dialect/IR/TcpOps.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/include/mlir-tcp/Dialect/IR/TcpOps.td -------------------------------------------------------------------------------- /include/mlir-tcp/Dialect/IR/TcpTilingInterfaceImpl.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/include/mlir-tcp/Dialect/IR/TcpTilingInterfaceImpl.h -------------------------------------------------------------------------------- /include/mlir-tcp/Dialect/IR/TcpTypes.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/include/mlir-tcp/Dialect/IR/TcpTypes.td -------------------------------------------------------------------------------- /include/mlir-tcp/Dialect/Transforms/DropSymbolicShapeOpsPass.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/include/mlir-tcp/Dialect/Transforms/DropSymbolicShapeOpsPass.h -------------------------------------------------------------------------------- /include/mlir-tcp/Dialect/Transforms/FuseTcpOpsPass.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/include/mlir-tcp/Dialect/Transforms/FuseTcpOpsPass.h -------------------------------------------------------------------------------- /include/mlir-tcp/Dialect/Transforms/FusionPatterns.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/include/mlir-tcp/Dialect/Transforms/FusionPatterns.h -------------------------------------------------------------------------------- /include/mlir-tcp/Dialect/Transforms/IsolateGroupOpsPass.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/include/mlir-tcp/Dialect/Transforms/IsolateGroupOpsPass.h -------------------------------------------------------------------------------- /include/mlir-tcp/Dialect/Transforms/Passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/include/mlir-tcp/Dialect/Transforms/Passes.h -------------------------------------------------------------------------------- /include/mlir-tcp/Dialect/Transforms/Passes.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/include/mlir-tcp/Dialect/Transforms/Passes.td -------------------------------------------------------------------------------- /include/mlir-tcp/Dialect/Transforms/TransformTensorOps.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/include/mlir-tcp/Dialect/Transforms/TransformTensorOps.h -------------------------------------------------------------------------------- /include/mlir-tcp/Dialect/Transforms/VerifyTcpBackendContractPass.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/include/mlir-tcp/Dialect/Transforms/VerifyTcpBackendContractPass.h -------------------------------------------------------------------------------- /include/mlir-tcp/InitAll.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/include/mlir-tcp/InitAll.h -------------------------------------------------------------------------------- /include/mlir-tcp/Pipeline/Pipeline.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/include/mlir-tcp/Pipeline/Pipeline.h -------------------------------------------------------------------------------- /lib/Conversion/PassDetail.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/lib/Conversion/PassDetail.h -------------------------------------------------------------------------------- /lib/Conversion/Passes.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/lib/Conversion/Passes.cpp -------------------------------------------------------------------------------- /lib/Conversion/StablehloToTcp/StablehloToTcp.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/lib/Conversion/StablehloToTcp/StablehloToTcp.cpp -------------------------------------------------------------------------------- /lib/Conversion/TcpToArith/TcpToArith.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/lib/Conversion/TcpToArith/TcpToArith.cpp -------------------------------------------------------------------------------- /lib/Conversion/TcpToLinalg/DataMovement.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/lib/Conversion/TcpToLinalg/DataMovement.cpp -------------------------------------------------------------------------------- /lib/Conversion/TcpToLinalg/Elementwise.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/lib/Conversion/TcpToLinalg/Elementwise.cpp -------------------------------------------------------------------------------- /lib/Conversion/TcpToLinalg/Misc.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/lib/Conversion/TcpToLinalg/Misc.cpp -------------------------------------------------------------------------------- /lib/Conversion/TcpToLinalg/PopulatePatterns.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/lib/Conversion/TcpToLinalg/PopulatePatterns.h -------------------------------------------------------------------------------- /lib/Conversion/TcpToLinalg/TcpToLinalg.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/lib/Conversion/TcpToLinalg/TcpToLinalg.cpp -------------------------------------------------------------------------------- /lib/Conversion/TcpToTensor/TcpToTensor.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/lib/Conversion/TcpToTensor/TcpToTensor.cpp -------------------------------------------------------------------------------- /lib/Conversion/TorchToTcp/DataMovement.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/lib/Conversion/TorchToTcp/DataMovement.cpp -------------------------------------------------------------------------------- /lib/Conversion/TorchToTcp/Elementwise.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/lib/Conversion/TorchToTcp/Elementwise.cpp -------------------------------------------------------------------------------- /lib/Conversion/TorchToTcp/Misc.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/lib/Conversion/TorchToTcp/Misc.cpp -------------------------------------------------------------------------------- /lib/Conversion/TorchToTcp/PopulatePatterns.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/lib/Conversion/TorchToTcp/PopulatePatterns.h -------------------------------------------------------------------------------- /lib/Conversion/TorchToTcp/TcpCustomOp.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/lib/Conversion/TorchToTcp/TcpCustomOp.cpp -------------------------------------------------------------------------------- /lib/Conversion/TorchToTcp/TorchToTcp.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/lib/Conversion/TorchToTcp/TorchToTcp.cpp -------------------------------------------------------------------------------- /lib/Conversion/TorchToTcp/TorchToTcpCustomOp.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/lib/Conversion/TorchToTcp/TorchToTcpCustomOp.cpp -------------------------------------------------------------------------------- /lib/Conversion/TorchToTcp/Utils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/lib/Conversion/TorchToTcp/Utils.cpp -------------------------------------------------------------------------------- /lib/Conversion/TorchToTcp/Utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/lib/Conversion/TorchToTcp/Utils.h -------------------------------------------------------------------------------- /lib/Dialect/IR/TcpDialect.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/lib/Dialect/IR/TcpDialect.cpp -------------------------------------------------------------------------------- /lib/Dialect/IR/TcpOps.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/lib/Dialect/IR/TcpOps.cpp -------------------------------------------------------------------------------- /lib/Dialect/IR/TcpTilingInterfaceImpl.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/lib/Dialect/IR/TcpTilingInterfaceImpl.cpp -------------------------------------------------------------------------------- /lib/Dialect/Transforms/DropSymbolicShapeOpsPass.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/lib/Dialect/Transforms/DropSymbolicShapeOpsPass.cpp -------------------------------------------------------------------------------- /lib/Dialect/Transforms/FuseTcpOpsPass.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/lib/Dialect/Transforms/FuseTcpOpsPass.cpp -------------------------------------------------------------------------------- /lib/Dialect/Transforms/FusionPatterns.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/lib/Dialect/Transforms/FusionPatterns.cpp -------------------------------------------------------------------------------- /lib/Dialect/Transforms/IsolateGroupOpsPass.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/lib/Dialect/Transforms/IsolateGroupOpsPass.cpp -------------------------------------------------------------------------------- /lib/Dialect/Transforms/PassDetail.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/lib/Dialect/Transforms/PassDetail.h -------------------------------------------------------------------------------- /lib/Dialect/Transforms/Passes.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/lib/Dialect/Transforms/Passes.cpp -------------------------------------------------------------------------------- /lib/Dialect/Transforms/TransformTensorOps.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/lib/Dialect/Transforms/TransformTensorOps.cpp -------------------------------------------------------------------------------- /lib/Dialect/Transforms/VerifyTcpBackendContractPass.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/lib/Dialect/Transforms/VerifyTcpBackendContractPass.cpp -------------------------------------------------------------------------------- /lib/InitAll.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/lib/InitAll.cpp -------------------------------------------------------------------------------- /lib/Pipeline/Pipeline.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/lib/Pipeline/Pipeline.cpp -------------------------------------------------------------------------------- /local_repos.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/local_repos.bzl -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/requirements.txt -------------------------------------------------------------------------------- /requirements_lock.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/requirements_lock.txt -------------------------------------------------------------------------------- /test/AotCompile/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/AotCompile/BUILD -------------------------------------------------------------------------------- /test/AotCompile/basic_tcp_ops.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/AotCompile/basic_tcp_ops.mlir -------------------------------------------------------------------------------- /test/AotCompile/basic_tcp_ops_compile_execute_test.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/AotCompile/basic_tcp_ops_compile_execute_test.cpp -------------------------------------------------------------------------------- /test/AotCompile/model_loader_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/AotCompile/model_loader_lib.py -------------------------------------------------------------------------------- /test/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/BUILD -------------------------------------------------------------------------------- /test/Conversion/StablehloToTcp/stablehlo.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/Conversion/StablehloToTcp/stablehlo.mlir -------------------------------------------------------------------------------- /test/Conversion/TcpToArith/basic.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/Conversion/TcpToArith/basic.mlir -------------------------------------------------------------------------------- /test/Conversion/TcpToLinalg/binary.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/Conversion/TcpToLinalg/binary.mlir -------------------------------------------------------------------------------- /test/Conversion/TcpToLinalg/data_movement.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/Conversion/TcpToLinalg/data_movement.mlir -------------------------------------------------------------------------------- /test/Conversion/TcpToLinalg/misc.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/Conversion/TcpToLinalg/misc.mlir -------------------------------------------------------------------------------- /test/Conversion/TcpToLinalg/unary.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/Conversion/TcpToLinalg/unary.mlir -------------------------------------------------------------------------------- /test/Conversion/TcpToTensor/data_movement.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/Conversion/TcpToTensor/data_movement.mlir -------------------------------------------------------------------------------- /test/Conversion/TorchToTcp/data_movement.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/Conversion/TorchToTcp/data_movement.mlir -------------------------------------------------------------------------------- /test/Conversion/TorchToTcp/dynamo_fx_import.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/Conversion/TorchToTcp/dynamo_fx_import.mlir -------------------------------------------------------------------------------- /test/Conversion/TorchToTcp/elementwise.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/Conversion/TorchToTcp/elementwise.mlir -------------------------------------------------------------------------------- /test/Conversion/TorchToTcp/misc.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/Conversion/TorchToTcp/misc.mlir -------------------------------------------------------------------------------- /test/Conversion/TorchToTcp/partial_conversion.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/Conversion/TorchToTcp/partial_conversion.mlir -------------------------------------------------------------------------------- /test/Conversion/TorchToTcp/tcp_custom_ops.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/Conversion/TorchToTcp/tcp_custom_ops.mlir -------------------------------------------------------------------------------- /test/Dialect/binary.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/Dialect/binary.mlir -------------------------------------------------------------------------------- /test/Dialect/canonicalize.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/Dialect/canonicalize.mlir -------------------------------------------------------------------------------- /test/Dialect/data_movement.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/Dialect/data_movement.mlir -------------------------------------------------------------------------------- /test/Dialect/decompose_tensor_ops.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/Dialect/decompose_tensor_ops.mlir -------------------------------------------------------------------------------- /test/Dialect/drop_symbolic_shape_ops.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/Dialect/drop_symbolic_shape_ops.mlir -------------------------------------------------------------------------------- /test/Dialect/misc.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/Dialect/misc.mlir -------------------------------------------------------------------------------- /test/Dialect/structured_fuse.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/Dialect/structured_fuse.mlir -------------------------------------------------------------------------------- /test/Dialect/tcp_custom_ops.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/Dialect/tcp_custom_ops.mlir -------------------------------------------------------------------------------- /test/Dialect/tcp_fusion.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/Dialect/tcp_fusion.mlir -------------------------------------------------------------------------------- /test/Dialect/tcp_isolate_groups.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/Dialect/tcp_isolate_groups.mlir -------------------------------------------------------------------------------- /test/Dialect/unary.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/Dialect/unary.mlir -------------------------------------------------------------------------------- /test/Pipeline/tcp_to_llvm_pipeline.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/Pipeline/tcp_to_llvm_pipeline.mlir -------------------------------------------------------------------------------- /test/Pipeline/torch_to_tcp_pipeline.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/Pipeline/torch_to_tcp_pipeline.mlir -------------------------------------------------------------------------------- /test/lit.cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/lit.cfg.py -------------------------------------------------------------------------------- /test/lit.site.cfg.py.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/lit.site.cfg.py.in -------------------------------------------------------------------------------- /test/python/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/python/BUILD -------------------------------------------------------------------------------- /test/python/fx_import/basic_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/python/fx_import/basic_test.py -------------------------------------------------------------------------------- /test/python_lit/fx_import/basic_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/python_lit/fx_import/basic_test.py -------------------------------------------------------------------------------- /test/python_lit/fx_import/custom_op_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/python_lit/fx_import/custom_op_test.py -------------------------------------------------------------------------------- /test/python_lit/fx_import/symbolic_shape_expr_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/test/python_lit/fx_import/symbolic_shape_expr_test.py -------------------------------------------------------------------------------- /third_party/.clang-format: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/third_party/.clang-format -------------------------------------------------------------------------------- /third_party/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/third_party/BUILD -------------------------------------------------------------------------------- /third_party/cnpy.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/third_party/cnpy.BUILD -------------------------------------------------------------------------------- /tools/aot/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/tools/aot/BUILD -------------------------------------------------------------------------------- /tools/aot/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/tools/aot/README.md -------------------------------------------------------------------------------- /tools/aot/abi.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/tools/aot/abi.h -------------------------------------------------------------------------------- /tools/aot/aot_compile.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/tools/aot/aot_compile.bzl -------------------------------------------------------------------------------- /tools/aot/execute_test.template.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/tools/aot/execute_test.template.cpp -------------------------------------------------------------------------------- /tools/aot/execute_test_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/tools/aot/execute_test_generator.py -------------------------------------------------------------------------------- /tools/aot/torch_exporter_harness.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/tools/aot/torch_exporter_harness.py -------------------------------------------------------------------------------- /tools/aot/torch_loader_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/tools/aot/torch_loader_utils.py -------------------------------------------------------------------------------- /tools/buildifier/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/tools/buildifier/BUILD -------------------------------------------------------------------------------- /tools/clangd/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/tools/clangd/BUILD -------------------------------------------------------------------------------- /tools/pip/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/tools/pip/BUILD -------------------------------------------------------------------------------- /tools/tcp-opt/tcp-opt.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cruise-automation/mlir-tcp/HEAD/tools/tcp-opt/tcp-opt.cpp --------------------------------------------------------------------------------