 AlpinDale
|
98f9dbd734
feat: Triton Kernels for Punica (#613)
|
hai 5 meses |
 AlpinDale
|
c978dfa1dd
fix: torch nightly version for rocm dockerfile
|
hai 5 meses |
|
AlpinDale
|
21c2a5ae46
chore: minor simplifications for Dockerfile.rocm
|
hai 5 meses |
|
AlpinDale
|
84a9cd25c9
fix: some naming issues
|
hai 5 meses |
|
AlpinDale
|
fa15bad2ea
chore: minor AMD fixes
|
hai 5 meses |
|
AlpinDale
|
0b2ae31122
cleanup rocm dockerfile
|
hai 5 meses |
|
AlpinDale
|
ddb28a80a3
fix: bump torch for rocm, unify CUDA_VISIBLE_DEVICES for cuda and rocm
|
hai 5 meses |
|
AlpinDale
|
aea0b52e52
fix: torchvision version for rocm
|
hai 5 meses |
|
AlpinDale
|
2c321ce1f2
chore: upgrade to rocm 6.1, update docker
|
hai 5 meses |
|
AlpinDale
|
89be49d058
fix: build for mi300x
|
hai 5 meses |
|
AlpinDale
|
ff13d47ef3
fix: copy commands in rocm docker
|
hai 5 meses |
|
AlpinDale
|
00acf371f9
rocm: fused topk softmax
|
hai 6 meses |
|
AlpinDale
|
f6250c5516
move dockerfiles to root; fix cpu build
|
hai 6 meses |
