Skip to content

GitLab

"torch_scatter/src/cpu.c" did not exist on "6e42a446ff51855b9776ded60fa0afea6ba5914f"
Switch branch/tag
Find file Normal viewHistoryPermalink
CMakeLists.txt 236 Bytes
Newer Older
carlushuang's avatar
introducing ck_tile! (#1216)  
carlushuang committed
1
2
3
4
5 
include_directories(AFTER
  ${CMAKE_CURRENT_LIST_DIR}
)

add_subdirectory(01_fmha)
rocking's avatar
layernorm2d forward (#1339)  
rocking committed
6 
add_subdirectory(02_layernorm2d)
Thomas Ning's avatar
Ck tile gemm example (#1488)  
Thomas Ning committed
7 
add_subdirectory(03_gemm)
Bartłomiej Kocot's avatar
[CK_TILE] Image to Column kernel (#1532)  
Bartłomiej Kocot committed
8 
add_subdirectory(04_img2col)
ltqin's avatar
update layernorm (#1570)  
ltqin committed
9 
add_subdirectory(05_reduce)
dummycoderfe's avatar
build ok  
dummycoderfe committed
10 
add_subdirectory(06_layernorm2d_bwd)