Wip/rocm sdk builder 612 bg91 - Githubissues

lamikr / rocm_sdk_builder

Other

132 stars 12 forks source link

Wip/rocm sdk builder 612 bg91 #92

Closed lamikr closed 3 months ago

lamikr commented 3 months ago

Builds latest upstream triton version and adds two tests with sligh modifications to original version to make them work with the latest pytorch and latest triton on rocm

docs/examples/pytorch/triton_migraph_quantization based on to https://github.com/ROCm/rocm-examples.git dir AI/MIGraphX/Quantization
docs/examples/pytorch/pytorch_gelu_vs_triton_gelu_kernel based on to https://rocm.blogs.amd.com/artificial-intelligence/triton/README.html

fixes https://github.com/lamikr/rocm_sdk_builder/issues/91

lamikr commented 3 months ago

Tested the added triton version with the examples on second commit of this pull request.