Skip to content

Commit

Permalink
Dockerfile.rocm.ubi: cleanup
Browse files Browse the repository at this point in the history
- get rid of non-essential dependencies
- consolidate package installs
- do not copy wheels in final stage
- fix ccache usage
- use flashattention with triton backend by default:
    - clone main_perf branch
    - build rocm target
    - set up triton rocm env var
- configure numba, outlines and triton cache directory
  • Loading branch information
dtrifiro committed Sep 16, 2024
1 parent 1ed711a commit 5af22d7
Showing 1 changed file with 119 additions and 172 deletions.
Loading

0 comments on commit 5af22d7

Please sign in to comment.