ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-06 05:55:39 +00:00
Code Issues Packages Projects Releases Wiki Activity
539 Commits 766 Branches 40 Tags
34c661e71cc8cf4753843a58786c8f6211ec5e22
Go to file
Clone
Open with VS Code Open with VSCodium Open with Intellij IDEA
Download ZIP Download TAR.GZ Download BUNDLE
Jianfeng Yan 34c661e71c Batched gemm and reduction (#156) 
* adding batched_gemm_and_reduction

* batched_gemm_reduce works with bactch_count=1

* fix a bug in grid_size; batched_gemm_reduce works for batch_count > 1

* adding profiler for batched_gemm_fp16

* fixed a bug in declaration of d1 and d0; both example and profiler work

* clang-format

* cleanup

* batched_gemm_reduce: add test

* minor change

* fixed some typo in function names
2022-03-30 11:21:18 -05:00
cmake
Initial Setup for CI (#86)
2022-02-18 21:44:11 -06:00
example
Batched gemm and reduction (#156)
2022-03-30 11:21:18 -05:00
external/include/half
Reorganize files, Part 1 (#119)
2022-03-08 21:46:36 -06:00
include/ck
Batched gemm and reduction (#156)
2022-03-30 11:21:18 -05:00
library
Batched gemm and reduction (#156)
2022-03-30 11:21:18 -05:00
profiler
Batched gemm and reduction (#156)
2022-03-30 11:21:18 -05:00
script
Unified conv3D API + support for all data types. (#133)
2022-03-23 10:23:13 -05:00
test
Batched gemm and reduction (#156)
2022-03-30 11:21:18 -05:00
.clang-format
start adding convolution
2018-10-08 22:49:58 -05:00
.clang-tidy
add tidy
2021-08-08 17:41:54 +00:00
.gitignore
Allow distinct K0/K1 values for A/B block descriptor (#98)
2022-02-27 21:06:18 -06:00
CMakeLists.txt
Reorganize files, Part 1 (#119)
2022-03-08 21:46:36 -06:00
dev-requirements.txt
Initial Setup for CI (#86)
2022-02-18 21:44:11 -06:00
Dockerfile
Reduction in Composable Kernel (#82)
2022-03-05 16:46:51 -06:00
Jenkinsfile
improve parallelism for testing (#112)
2022-03-07 10:33:12 -06:00
rbuild.ini
Update test CMakeLists to add new tests automatically and add Jenkins stage for tests (#88)
2022-03-03 16:59:42 -06:00
README.md
ckProfiler and device-level XDL GEMM operator (#48)
2021-11-14 11:28:32 -06:00
requirements.txt
Update test CMakeLists to add new tests automatically and add Jenkins stage for tests (#88)
2022-03-03 16:59:42 -06:00

README.md

Reference in New Issue View Git Blame Copy Permalink
Description
[DEPRECATED] Moved to ROCm/rocm-libraries repo. NOTE: develop branch is maintained as a read-only mirror
Readme MIT 249 MiB
Languages
C++ 92%
Python 5.5%
CMake 1.5%
Shell 0.5%
Pawn 0.2%