"...composable_kernel_rocm.git" did not exist on "b94fd0b2279c6476c6e109e99dc5d0e6d8ce313c"
Commit a6ea41de authored by Chao Liu's avatar Chao Liu
Browse files

update examle

parent 4746e11b
......@@ -215,13 +215,13 @@ int main(int argc, char* argv[])
}
// A[M0, M1, K0, K1]
std::vector<ck::index_t> a_ms_ks_lengths{30, 128, 32, 128};
std::vector<ck::index_t> a_ms_ks_lengths{30, 128, 16, 128};
std::vector<ck::index_t> a_ms_ks_strides{524288, 4096, 128, 1};
// B[K0, K1, N0, N1]
std::vector<ck::index_t> b_ks_ns_lengths{32, 128, 32, 128};
std::vector<ck::index_t> b_ks_ns_lengths{16, 128, 16, 128};
std::vector<ck::index_t> b_ks_ns_strides{128, 1, 524288, 4096};
// C[M0, M1, N0, N1]
std::vector<ck::index_t> c_ms_ns_lengths{30, 128, 32, 128};
std::vector<ck::index_t> c_ms_ns_lengths{30, 128, 16, 128};
std::vector<ck::index_t> c_ms_ns_strides{524288, 4096, 128, 1};
Tensor<ADataType> a_ms_ks(
......
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