CMakeLists.txt 360 字节
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14
if(WITH_GPU)
  nv_library(
    phi_place
    SRCS place.cc
    DEPS phi_gpu_info)
elseif(WITH_ROCM)
  hip_library(
    phi_place
    SRCS place.cc
    DEPS phi_gpu_info)
else()
  cc_library(phi_place SRCS place.cc)
endif()

15 16 17 18 19 20 21 22
cc_library(
  scalar
  SRCS scalar.cc
  DEPS phi_enforce tensor)
cc_library(
  int_array
  SRCS int_array.cc
  DEPS phi_enforce tensor)