mirror of
https://github.com/ROCm/ROCm.git
synced 2026-02-21 03:00:39 -05:00
e5d7bb4fae
rename tl.float8e4 to tl.float8e4nv to align with upstream ROCM IFU: Fix python arch issues ROCM IFU: Fix kernel launcher ROCM IFU: Fix merge conflicts fix debug build Set correct threadsPerCTA
31 lines
904 B
Bash
Executable File
31 lines
904 B
Bash
Executable File
set -o xtrace
|
|
|
|
DRUN='sudo docker run -it --rm --network=host --user root --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined'
|
|
|
|
# DEVICES="--gpus all"
|
|
DEVICES="--device=/dev/kfd --device=/dev/dri"
|
|
|
|
MEMORY="--ipc=host --shm-size 16G"
|
|
|
|
VOLUMES="-v $HOME/dockerx:/dockerx -v /data:/data"
|
|
|
|
# WORK_DIR='/root/$(basename $(pwd))'
|
|
WORK_DIR="/dockerx/$(basename $(pwd))"
|
|
|
|
IMAGE_NAME=rocm/pytorch-nightly:latest
|
|
# IMAGE_NAME=rocm/pytorch:latest
|
|
# IMAGE_NAME=nvcr.io/nvidia/pytorch
|
|
|
|
CONTAINER_NAME=triton
|
|
|
|
# start new container
|
|
docker stop $CONTAINER_NAME
|
|
docker rm $CONTAINER_NAME
|
|
CONTAINER_ID=$($DRUN -d -w $WORK_DIR --name $CONTAINER_NAME $MEMORY $VOLUMES $DEVICES $IMAGE_NAME)
|
|
echo "CONTAINER_ID: $CONTAINER_ID"
|
|
# docker cp . $CONTAINER_ID:$WORK_DIR
|
|
# docker exec $CONTAINER_ID bash -c "bash scripts/amd/run.sh"
|
|
docker attach $CONTAINER_ID
|
|
docker stop $CONTAINER_ID
|
|
docker rm $CONTAINER_ID
|