Skip to content

P
Paddle

机器未来 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
切换分支/标签
查找文件 普通视图历史永久链接Permalink
CMakeLists.txt 323 字节
Newer Older
C
adjust main dir (#37916)  
Chen Weihang 已提交
1 2 
add_subdirectory(eigen)
C
[PTen] Unify device context entrance in pten part 2 (#38182)  
Chen Weihang 已提交
3 
cc_library(pten_transpose_cpu SRCS transpose.cc DEPS dense_tensor pten_context)
C
[Pten] Add reduce mean kernel, replace with mean API (#37559)  
chentianyu03 已提交
4 
if(WITH_GPU)
C
[PTen] Rename cuda dir and context to gpu (#38296)  
Chen Weihang 已提交
5 
  nv_library(pten_transpose_gpu SRCS transpose.cu DEPS dense_tensor malloc pten_context)
C
[Pten] Add reduce mean kernel, replace with mean API (#37559)  
chentianyu03 已提交
6 
elseif(WITH_ROCM)
C
[PTen] Rename cuda dir and context to gpu (#38296)  
Chen Weihang 已提交
7 
  hip_library(pten_transpose_gpu SRCS transpose.cu DEPS dense_tensor malloc pten_context)
C
[Pten] Add reduce mean kernel, replace with mean API (#37559)  
chentianyu03 已提交
8 
endif()