Fix Tensorflow build with CUDA 10.1
@@ -165,7 +165,13 @@ ENDIF()
IF (HOST_OS_WINDOWS)
LDFLAGS(cublas.lib curand.lib cudart_static.lib cusparse.lib)
ELSE()
- EXTRALIBS(-lcublas_static -lcurand_static -lcudart_static -lcusparse_static -lculibos)
+ EXTRALIBS(-lcurand_static -lcudart_static -lcusparse_static -lculibos)
+ IF (CUDA_VERSION STREQUAL "10.1")
+ FROM_SANDBOX(1006791015 OUT cublas.a)
+ EXTRALIBS(-llapack_static)
+ ELSE()
+ EXTRALIBS(-lcublas_static)
+ ENDIF()
ENDIF()
END()
