mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-11 17:00:18 +00:00
b3e8d57d51
* add parameters * tweak gemm * tweak * update conv * update script * adding bwd 1x1 * update script * adding 1x1 bwd * debugging bwd 1x1 failure * update script * update script * test * test v100 * clean up
15 lines
767 B
Bash
Executable File
15 lines
767 B
Bash
Executable File
WORKSPACE=$1
|
|
echo "workspace: " $WORKSPACE
|
|
|
|
docker run \
|
|
-it \
|
|
--rm \
|
|
--privileged \
|
|
--group-add sudo \
|
|
-w /root/workspace \
|
|
-v $WORKSPACE:/root/workspace \
|
|
rocm/tensorflow:rocm4.3.1-tf2.6-dev \
|
|
/bin/bash
|
|
|
|
#--network host \
|