m53297601
/
Paddle
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from '5f1081d83d'
${ noResults }
Paddle/paddle/operators/math/CMakeLists.txt

17 lines
564 B
Raw Blame History

if (WITH_GPU)
if (WITH_MKLML)
nv_library(math_function SRCS math_function.cc math_function.cu DEPS mklml device_context)
else()
nv_library(math_function SRCS math_function.cc math_function.cu DEPS cblas device_context)
endif()
else()
if (WITH_MKLML)
cc_library(math_function SRCS math_function.cc DEPS mklml device_context)
else()
cc_library(math_function SRCS math_function.cc DEPS cblas device_context)
endif()
endif()
nv_test(math_function_test SRCS math_function_test.cc DEPS math_function tensor)
Reference in new issue View Git Blame Copy Permalink