├── .ci ├── Dockerfile └── README.md ├── .github └── workflows │ ├── autotest.yml │ └── dockerbuild.yml ├── .gitignore ├── .gitmodules ├── LICENSE ├── PolyTool - LLVM Conference 2025.pdf ├── PyDSL - LLVM Conference 2024.pdf ├── PyDSL - LLVM Conference 2025.pdf ├── README.md ├── dev ├── build.md ├── dev_docs.md └── print_ast.py ├── docs ├── changelog.md ├── issues.md └── usage.md ├── examples ├── affine_explicit.py ├── affine_implicit.py ├── autotune │ ├── heat_demo.py │ ├── tune_combined.py │ ├── tune_program.py │ ├── tune_transform.py │ └── tune_type.py ├── function.py ├── hello_memref.py ├── memref.py ├── mlp.py ├── simple.py ├── triton_interop.py ├── tuple.py └── with_recursively.py ├── pyproject.toml ├── requirements.txt ├── src └── pydsl │ ├── __init__.py │ ├── affine.py │ ├── analysis │ ├── __init__.py │ ├── dataflow.py │ └── names.py │ ├── arith.py │ ├── attribute.py │ ├── autotune.py │ ├── compiler.py │ ├── frontend.py │ ├── func.py │ ├── gpu.py │ ├── linalg.py │ ├── llvm.py │ ├── macro.py │ ├── math.py │ ├── memref.py │ ├── mesh.py │ ├── protocols.py │ ├── scf.py │ ├── scope.py │ ├── tensor.py │ ├── transform.py │ ├── triton.py │ ├── type.py │ └── vector.py └── tests ├── __init__.py ├── e2e ├── __init__.py ├── test_affine.py ├── test_algorithms.py ├── test_arith.py ├── test_error_msg.py ├── test_examples.py ├── test_frontend.py ├── test_func.py ├── test_linalg.py ├── test_macro.py ├── test_memref.py ├── test_scf.py ├── test_syntax.py ├── test_templates.py ├── test_tensor.py ├── test_transform.py ├── test_triton.py └── test_tuple.py ├── helper.py └── unit ├── __init__.py ├── test_frontend.py └── test_protocols.py /.ci/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/.ci/Dockerfile -------------------------------------------------------------------------------- /.ci/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/.ci/README.md -------------------------------------------------------------------------------- /.github/workflows/autotest.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/.github/workflows/autotest.yml -------------------------------------------------------------------------------- /.github/workflows/dockerbuild.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/.github/workflows/dockerbuild.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/.gitmodules -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/LICENSE -------------------------------------------------------------------------------- /PolyTool - LLVM Conference 2025.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/PolyTool - LLVM Conference 2025.pdf -------------------------------------------------------------------------------- /PyDSL - LLVM Conference 2024.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/PyDSL - LLVM Conference 2024.pdf -------------------------------------------------------------------------------- /PyDSL - LLVM Conference 2025.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/PyDSL - LLVM Conference 2025.pdf -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/README.md -------------------------------------------------------------------------------- /dev/build.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/dev/build.md -------------------------------------------------------------------------------- /dev/dev_docs.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/dev/dev_docs.md -------------------------------------------------------------------------------- /dev/print_ast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/dev/print_ast.py -------------------------------------------------------------------------------- /docs/changelog.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/docs/changelog.md -------------------------------------------------------------------------------- /docs/issues.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/docs/issues.md -------------------------------------------------------------------------------- /docs/usage.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/docs/usage.md -------------------------------------------------------------------------------- /examples/affine_explicit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/examples/affine_explicit.py -------------------------------------------------------------------------------- /examples/affine_implicit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/examples/affine_implicit.py -------------------------------------------------------------------------------- /examples/autotune/heat_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/examples/autotune/heat_demo.py -------------------------------------------------------------------------------- /examples/autotune/tune_combined.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/examples/autotune/tune_combined.py -------------------------------------------------------------------------------- /examples/autotune/tune_program.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/examples/autotune/tune_program.py -------------------------------------------------------------------------------- /examples/autotune/tune_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/examples/autotune/tune_transform.py -------------------------------------------------------------------------------- /examples/autotune/tune_type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/examples/autotune/tune_type.py -------------------------------------------------------------------------------- /examples/function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/examples/function.py -------------------------------------------------------------------------------- /examples/hello_memref.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/examples/hello_memref.py -------------------------------------------------------------------------------- /examples/memref.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/examples/memref.py -------------------------------------------------------------------------------- /examples/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/examples/mlp.py -------------------------------------------------------------------------------- /examples/simple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/examples/simple.py -------------------------------------------------------------------------------- /examples/triton_interop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/examples/triton_interop.py -------------------------------------------------------------------------------- /examples/tuple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/examples/tuple.py -------------------------------------------------------------------------------- /examples/with_recursively.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/examples/with_recursively.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/pydsl/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/pydsl/affine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/src/pydsl/affine.py -------------------------------------------------------------------------------- /src/pydsl/analysis/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/pydsl/analysis/dataflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/src/pydsl/analysis/dataflow.py -------------------------------------------------------------------------------- /src/pydsl/analysis/names.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/src/pydsl/analysis/names.py -------------------------------------------------------------------------------- /src/pydsl/arith.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/src/pydsl/arith.py -------------------------------------------------------------------------------- /src/pydsl/attribute.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/src/pydsl/attribute.py -------------------------------------------------------------------------------- /src/pydsl/autotune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/src/pydsl/autotune.py -------------------------------------------------------------------------------- /src/pydsl/compiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/src/pydsl/compiler.py -------------------------------------------------------------------------------- /src/pydsl/frontend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/src/pydsl/frontend.py -------------------------------------------------------------------------------- /src/pydsl/func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/src/pydsl/func.py -------------------------------------------------------------------------------- /src/pydsl/gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/src/pydsl/gpu.py -------------------------------------------------------------------------------- /src/pydsl/linalg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/src/pydsl/linalg.py -------------------------------------------------------------------------------- /src/pydsl/llvm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/src/pydsl/llvm.py -------------------------------------------------------------------------------- /src/pydsl/macro.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/src/pydsl/macro.py -------------------------------------------------------------------------------- /src/pydsl/math.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/src/pydsl/math.py -------------------------------------------------------------------------------- /src/pydsl/memref.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/src/pydsl/memref.py -------------------------------------------------------------------------------- /src/pydsl/mesh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/src/pydsl/mesh.py -------------------------------------------------------------------------------- /src/pydsl/protocols.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/src/pydsl/protocols.py -------------------------------------------------------------------------------- /src/pydsl/scf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/src/pydsl/scf.py -------------------------------------------------------------------------------- /src/pydsl/scope.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/src/pydsl/scope.py -------------------------------------------------------------------------------- /src/pydsl/tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/src/pydsl/tensor.py -------------------------------------------------------------------------------- /src/pydsl/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/src/pydsl/transform.py -------------------------------------------------------------------------------- /src/pydsl/triton.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/src/pydsl/triton.py -------------------------------------------------------------------------------- /src/pydsl/type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/src/pydsl/type.py -------------------------------------------------------------------------------- /src/pydsl/vector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/src/pydsl/vector.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/e2e/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/e2e/test_affine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/tests/e2e/test_affine.py -------------------------------------------------------------------------------- /tests/e2e/test_algorithms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/tests/e2e/test_algorithms.py -------------------------------------------------------------------------------- /tests/e2e/test_arith.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/tests/e2e/test_arith.py -------------------------------------------------------------------------------- /tests/e2e/test_error_msg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/tests/e2e/test_error_msg.py -------------------------------------------------------------------------------- /tests/e2e/test_examples.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/tests/e2e/test_examples.py -------------------------------------------------------------------------------- /tests/e2e/test_frontend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/tests/e2e/test_frontend.py -------------------------------------------------------------------------------- /tests/e2e/test_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/tests/e2e/test_func.py -------------------------------------------------------------------------------- /tests/e2e/test_linalg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/tests/e2e/test_linalg.py -------------------------------------------------------------------------------- /tests/e2e/test_macro.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/tests/e2e/test_macro.py -------------------------------------------------------------------------------- /tests/e2e/test_memref.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/tests/e2e/test_memref.py -------------------------------------------------------------------------------- /tests/e2e/test_scf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/tests/e2e/test_scf.py -------------------------------------------------------------------------------- /tests/e2e/test_syntax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/tests/e2e/test_syntax.py -------------------------------------------------------------------------------- /tests/e2e/test_templates.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/tests/e2e/test_templates.py -------------------------------------------------------------------------------- /tests/e2e/test_tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/tests/e2e/test_tensor.py -------------------------------------------------------------------------------- /tests/e2e/test_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/tests/e2e/test_transform.py -------------------------------------------------------------------------------- /tests/e2e/test_triton.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/tests/e2e/test_triton.py -------------------------------------------------------------------------------- /tests/e2e/test_tuple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/tests/e2e/test_tuple.py -------------------------------------------------------------------------------- /tests/helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/tests/helper.py -------------------------------------------------------------------------------- /tests/unit/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/unit/test_frontend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/tests/unit/test_frontend.py -------------------------------------------------------------------------------- /tests/unit/test_protocols.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huawei-CPLLab/PyDSL/HEAD/tests/unit/test_protocols.py --------------------------------------------------------------------------------