Commit 2593dd60 authored by turneram's avatar turneram
Browse files

Formatting

parent d1e27426
......@@ -61,8 +61,8 @@ namespace gpu {
// // ck_gemm(xs...);
// // });
// make_tensors()(a_p, b_p, c_p)([](auto a_t, auto b_t, auto c_t) {
// __shared__ float p_shared_block[512]; //[(a_t.get_shape().elements() + b_t.get_shape().elements()) * 2];
// ck_gemm(a_t, b_t, c_t, p_shared_block);
// __shared__ float p_shared_block[512]; //[(a_t.get_shape().elements() +
// b_t.get_shape().elements()) * 2]; ck_gemm(a_t, b_t, c_t, p_shared_block);
// // make_tensors()(p_shared_block)([&](auto p_t) {
// // ck_gemm(a_t, b_t, c_t, p_t);
// // });
......
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