Skip to content

P
Paddle

OPTHREE / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle

体验新版 GitCode，发现更多精彩内容 >>

切换分支/标签
查找文件 普通视图历史永久链接Permalink
CMakeLists.txt 308 字节
Newer Older
C
[PTen] Add register_ctx_kernel marco and move scale kernel (#38121)  
Chen Weihang 已提交
1 2 3 4 5 6 7 
if(WITH_GPU)
  nv_library(scale_kernel_eigen SRCS scale_kernel.cc DEPS dense_tensor eigen_function)
elseif(WITH_ROCM)
  hip_library(scale_kernel_eigen SRCS scale_kernel.cc DEPS dense_tensor eigen_function)
else()
  cc_library(scale_kernel_eigen SRCS scale_kernel.cc DEPS dense_tensor eigen_function)
endif()