Unverified Commit 90437264 authored by zjing14's avatar zjing14 Committed by GitHub
Browse files

Update example/15_grouped_gemm/grouped_gemm_multi_abd_xdl_fixed_nk_bias_fp16.cpp


Co-authored-by: default avatarBartłomiej Kocot <barkocot@amd.com>
parent 62d65610
...@@ -249,10 +249,10 @@ bool run_grouped_gemm(const ProblemSize& problem_size, const ExecutionConfig& co ...@@ -249,10 +249,10 @@ bool run_grouped_gemm(const ProblemSize& problem_size, const ExecutionConfig& co
1}); 1});
grouped_gemm_kernel_args_.push_back( grouped_gemm_kernel_args_.push_back(
{std::array<const void*, 2>{a0_tensors_device[i]->GetDeviceBuffer(), {std::array<const void*, NumATensor>{a0_tensors_device[i]->GetDeviceBuffer(),
a1_tensors_device[i]->GetDeviceBuffer()}, a1_tensors_device[i]->GetDeviceBuffer()},
std::array<const void*, 1>{b_tensors_device[i]->GetDeviceBuffer()}, std::array<const void*, NumBTensor>{b_tensors_device[i]->GetDeviceBuffer()},
std::array<const void*, 1>{d0_tensors_device[i]->GetDeviceBuffer()}, std::array<const void*, NumDTensor>{d0_tensors_device[i]->GetDeviceBuffer()},
c_tensors_device[i]->GetDeviceBuffer(), c_tensors_device[i]->GetDeviceBuffer(),
problem_size.Ms[i], problem_size.Ms[i],
problem_size.Ns[i], problem_size.Ns[i],
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment