AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

adding mytest target

Browse Source 
Change-Id: Ia82b971597a4876d99c85d7dbbf997faea83f0c9
This commit is contained in:
Evgeny
2020-03-19 01:27:06 -05:00
parent f69c938f52
commit ffa0a51408
4 changed files with 56 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files
bin/build_kernel.sh
+28 -9
View File
@@ -1,7 +1,9 @@
#!/bin/sh
#!/bin/sh -x
TEST_NAME=$1
DST_DIR=$2
ROCM_DIR=$3
TGT_LIST=$4
if [ -z "$TEST_NAME" ] ; then
echo "Usage: $0 <test name> <dst dir>"
@@ -13,18 +15,35 @@ if [ -z "$DST_DIR" ] ; then
DST_DIR=$(dirname TEST_NAME)
fi
GFXIP=$(/opt/rocm/bin/rocminfo | grep "amdgcn-amd-amdhsa--" | head -n 1 | sed -n "s/^.*amdgcn-amd-amdhsa--\(\w*\).*$/\1/p")
if [ -z "$GFXIP" ] ; then
echo "GPU is not found"
if [ -z "$ROCM_DIR" ] ; then
ROCM_DIR=/opt/rocm
fi
if [ -z "$TGT_LIST" ] ; then
TGT_LIST=$(/opt/rocm/bin/rocminfo | grep "amdgcn-amd-amdhsa--" | head -n 1 | sed -n "s/^.*amdgcn-amd-amdhsa--\(\w*\).*$/\1/p")
fi
if [ -z "$TGT_LIST" ] ; then
echo "Error: GPU targets not found"
exit 1
fi
OBJ_PREF=$(echo $GFXIP | head -c 4)
OBJ_NAME=$(echo "_$(basename $TEST_NAME)" | sed -e 's/_./\U&\E/g' -e 's/_//g')
OBJ_FILE=${OBJ_PREF}_${OBJ_NAME}.hsaco
OCL_VER="2.0"
OCL_DIR=$ROCM_DIR/opencl
/opt/rocm/opencl/bin/x86_64/clang -cl-std=CL2.0 -cl-std=CL2.0 -include /opt/rocm/opencl/include/opencl-c.h -Xclang -mlink-bitcode-file -Xclang /opt/rocm/opencl/lib/x86_64/bitcode/opencl.amdgcn.bc -Xclang -mlink-bitcode-file -Xclang /opt/rocm/opencl/lib/x86_64/bitcode/ockl.amdgcn.bc -target amdgcn-amd-amdhsa -mcpu=$GFXIP -mno-code-object-v3 $TEST_NAME.cl -o $OBJ_FILE
LLVM_DIR=$ROCM_DIR/hcc
CLANG=$LLVM_DIR/bin/clang
BITCODE_OPTS="\
-Xclang -mlink-bitcode-file -Xclang $LLVM_DIR/lib/opencl.amdgcn.bc \
-Xclang -mlink-bitcode-file -Xclang $LLVM_DIR/lib/ockl.amdgcn.bc \
-Xclang -mlink-bitcode-file -Xclang $LLVM_DIR/lib/ocml.amdgcn.bc"
echo "'$OBJ_FILE' is generated for '$GFXIP'"
for GFXIP in $TGT_LIST ; do
OBJ_PREF=$GFXIP
OBJ_NAME=$(echo "_$(basename $TEST_NAME)" | sed -e 's/_./\U&\E/g' -e 's/_//g')
OBJ_FILE=${OBJ_PREF}_${OBJ_NAME}.hsaco
$CLANG -cl-std=CL$OCL_VER -include $OCL_DIR/include/opencl-c.h $BITCODE_OPTS -target amdgcn-amd-amdhsa -mcpu=$GFXIP -mno-code-object-v3 $TEST_NAME.cl -o $DST_DIR/$OBJ_FILE
echo "'$OBJ_FILE' is generated for '$GFXIP'"
done
exit 0