fix: rocblas arch
This commit is contained in:
+1
-1
@@ -82,7 +82,7 @@ RUN git clone -b ${ROCBLAS_BRANCH} https://github.com/ROCmSoftwarePlatform/rocBL
|
|||||||
WORKDIR /root/rocBLAS
|
WORKDIR /root/rocBLAS
|
||||||
RUN rm -rf library/src/blas3/Tensile/Logic/asm_full/r9nano/
|
RUN rm -rf library/src/blas3/Tensile/Logic/asm_full/r9nano/
|
||||||
RUN sed -i -e 's/rmake.py --install/rmake.py -a="${ROCBLAS_ARCH}" --install/g' install.sh
|
RUN sed -i -e 's/rmake.py --install/rmake.py -a="${ROCBLAS_ARCH}" --install/g' install.sh
|
||||||
RUN ./install.sh -d --cmake_install
|
RUN ./install.sh -a ${ROCBLAS_ARCH} -d --cmake_install
|
||||||
WORKDIR /root/rocBLAS/build/release
|
WORKDIR /root/rocBLAS/build/release
|
||||||
COPY fix-rocblas.sh .
|
COPY fix-rocblas.sh .
|
||||||
RUN <<EOF
|
RUN <<EOF
|
||||||
|
|||||||
Reference in New Issue
Block a user