diff --git a/bin/yaml/libraries.yaml b/bin/yaml/libraries.yaml index b297c17e..e4d1db75 100644 --- a/bin/yaml/libraries.yaml +++ b/bin/yaml/libraries.yaml @@ -1,4 +1,12 @@ libraries: + triton: + triton-shared: + type: tarballs + dir: triton-shared-{name} + compression: gz + url: https://github.com/microsoft/triton-shared/archive/refs/heads/main.tar.gz + targets: + - 'main' android-java: android-api-stubs: check_file: android.jar diff --git a/bin/yaml/tools.yaml b/bin/yaml/tools.yaml index 1ac13e6c..aaa0a9c0 100644 --- a/bin/yaml/tools.yaml +++ b/bin/yaml/tools.yaml @@ -98,6 +98,22 @@ tools: check_exe: bin/osaca --version targets: - 0.5.2 + wheel: + type: pip + dir: wheel-{name} + python: /usr/bin/python3 + package: wheel=={name} + check_exe: wheel version + targets: + - 0.43.0 + pybind11: + type: pip + dir: pybind11-{name} + python: /usr/bin/python3 + package: pybind11=={name} + check_exe: bin/pybind11-config --version + targets: + - 2.13.5 rustfmt: type: tarballs dir: rustfmt-{name} diff --git a/bin/yaml/triton.yaml b/bin/yaml/triton.yaml new file mode 100644 index 00000000..3ddc4714 --- /dev/null +++ b/bin/yaml/triton.yaml @@ -0,0 +1,22 @@ +compilers: + triton: + type: tarballs + dir: triton-{name} + depends: + - tools/cmake 3.29.2 + - tools/ninja 1.11.1 + - tools/wheel 0.43.0 + - tools/pybind11 2.13.5 + - compilers/c++/clang 18.1.0 + - libraries/triton/triton-shared main + compression: gz + check_exe: python3 venv/lib/python3.12/site-packages/triton/tools/compile.py -h + url: https://github.com/triton-lang/triton/archive/refs/tags/v{name}.tar.gz + targets: + - 2.1.0 + after_stage_script: + - cd triton-{name} + - python3 -m venv venv + - source venv/bin/activate + - cd python + - TRITON_BUILD_WITH_CLANG_LLD=true TRITON_PLUGIN_DIRS=/opt/compiler-explorer/triton-shared-main pip install -e . \ No newline at end of file