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

Formatting

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