 AlpinDale
|
ddb28a80a3
fix: bump torch for rocm, unify CUDA_VISIBLE_DEVICES for cuda and rocm
|
6 meses atrás |
|
AlpinDale
|
aea0b52e52
fix: torchvision version for rocm
|
7 meses atrás |
|
AlpinDale
|
2c321ce1f2
chore: upgrade to rocm 6.1, update docker
|
7 meses atrás |
|
AlpinDale
|
89be49d058
fix: build for mi300x
|
7 meses atrás |
|
AlpinDale
|
ff13d47ef3
fix: copy commands in rocm docker
|
7 meses atrás |
|
AlpinDale
|
00acf371f9
rocm: fused topk softmax
|
7 meses atrás |
|
AlpinDale
|
f6250c5516
move dockerfiles to root; fix cpu build
|
7 meses atrás |
