CMakeLists.txt 360 字节
Newer Older
1 2 3 4
if(WITH_GPU)
  nv_library(
    phi_place
    SRCS place.cc
5
    DEPS phi_backends)
6 7 8 9
elseif(WITH_ROCM)
  hip_library(
    phi_place
    SRCS place.cc
10
    DEPS phi_backends)
11 12 13 14
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)