Skip to content

Commit 48f860c

Browse files
add template function of hip and cmakelists.txt
1 parent 94099e7 commit 48f860c

2 files changed

Lines changed: 394 additions & 316 deletions

File tree

Lines changed: 44 additions & 43 deletions
Original file line numberDiff line numberDiff line change
@@ -1,44 +1,45 @@
1-
# list(APPEND module_device_srcs
2-
# memory_op.cpp
3-
# device.cpp
4-
# )
5-
6-
# if(USE_CUDA)
7-
# list(APPEND module_device_srcs
8-
# cuda/memory_op.cu
9-
# )
10-
# endif()
11-
12-
13-
# if(USE_ROCM)
14-
# hip_add_library(module_device_rocm STATIC
15-
# rocm/memory_op.hip.cu
16-
# )
17-
# target_link_libraries(
18-
# device
19-
# module_device_rocm
20-
# hip::host
21-
# hip::device
22-
# hip::hipfft
23-
# roc::hipblas
24-
# roc::hipsolver
25-
# )
26-
# endif()
27-
28-
# add_library(device OBJECT ${device_srcs})
29-
30-
# if(USE_CUDA)
31-
# target_link_libraries(
32-
# device
33-
# )
34-
# elseif(USE_ROCM)
35-
# target_link_libraries(
36-
# device
37-
# device_rocm
38-
# hip::host
39-
# hip::device
40-
# hip::hipfft
41-
# roc::hipblas
42-
# roc::hipsolver
43-
# )
1+
# list(APPEND module_device_srcs
2+
# memory_op.cpp
3+
# device.cpp
4+
# )
5+
6+
# if(USE_CUDA)
7+
# list(APPEND module_device_srcs
8+
# cuda/memory_op.cu
9+
# )
10+
# endif()
11+
12+
# add_library(device OBJECT ${module_device_srcs})
13+
14+
# if(USE_ROCM)
15+
# hip_add_library(module_device_rocm STATIC
16+
# rocm/memory_op.hip.cu
17+
# )
18+
# target_link_libraries(
19+
# device
20+
# module_device_rocm
21+
# hip::host
22+
# hip::device
23+
# hip::hipfft
24+
# roc::hipblas
25+
# roc::hipsolver
26+
# )
27+
# endif()
28+
29+
30+
31+
# if(USE_CUDA)
32+
# target_link_libraries(
33+
# device
34+
# )
35+
# elseif(USE_ROCM)
36+
# target_link_libraries(
37+
# device
38+
# device_rocm
39+
# hip::host
40+
# hip::device
41+
# hip::hipfft
42+
# roc::hipblas
43+
# roc::hipsolver
44+
# )
4445
# endif()

0 commit comments

Comments
 (0)