ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-13 01:36:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
529 Commits 742 Branches 38 Tags
d91f9f119c167ac5f3974e78f09bdd007f5dfd4a
Commit Graph

3 Commits

Author SHA1 Message Date
Jianfeng Yan
d91f9f119c Batched gemm bf16 (#142) 
* add bf16 for batched gemm

* batched_gemm_bf16 works

* recover accidently changed files
 2022-03-22 18:18:43 -05:00
Jianfeng Yan
cb87b049de refactored deviceBatchedGemm; removed GridwiseBatchedGemm; added fp32 and int8 to profiler (#120) 
changed long_index_t to index_t when computing memory offset

uncomment other ops in profiler

added test for batched_gemm
 2022-03-21 16:45:14 -05:00
zjing14
b53e9d08ed Batched GEMM for fp16 (#79) 
* prepare host for batched_gemm

* init commit of batched kernels

* fixed

* refine transform with freeze

* m/n padding

* fixed a bug; clean

* add small tiles

* clean

* clean code

* clean code

* add nt, tn, tt layout

* add missing file

* use StaticBufferTupleOfVector instead

* add reference_batched_gemm

* fixed a macro
 2022-02-11 09:36:52 -06:00