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 12, 2024
1 parent 3bd5180 commit b40ecd9
Showing 1 changed file with 122 additions and 172 deletions.
Loading

0 comments on commit b40ecd9

Please sign in to comment.