search

ROCm / triton

Development repository for the Triton language and compiler
MIT License
80 stars 23 forks source link

Dockerfile and test #458

Closed micmelesse closed 6 months ago

micmelesse commented 6 months ago

This PR add builds a docker file that pulls upstream and tests it on a AMD node.

You can build and run unit tests in the dockerfile with the following.

docker build -t triton .

sudo docker run -it --rm --network=host --user root --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined --ipc=host --shm-size 16G --device=/dev/kfd --device=/dev/dri \
    -w /tmp/triton --name triton_container -d triton

docker exec triton_container pytest -n 32 --capture=tee-sys -rfs --verbose "python/test/unit/language/test_core.py"

docker stop triton_container