4 Commits

Author	SHA1	Message	Date
Jianfeng Yan	87522d2ac9	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 [ROCm/composable_kernel commit: 34c661e71c]	2022-03-30 11:21:18 -05:00
Jianfeng Yan	225c98244d	Batched gemm bf16 (#142) ... * add bf16 for batched gemm * batched_gemm_bf16 works * recover accidently changed files [ROCm/composable_kernel commit: d91f9f119c]	2022-03-22 18:18:43 -05:00
Jianfeng Yan	aa4c28d53b	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 [ROCm/composable_kernel commit: cb87b049de]	2022-03-21 16:45:14 -05:00
zjing14	4795d9803d	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 [ROCm/composable_kernel commit: b53e9d08ed]	2022-02-11 09:36:52 -06:00