Commit 30b3276d authored by Shucai Xiao's avatar Shucai Xiao
Browse files

clang format

parent c0d3babb
...@@ -40,11 +40,11 @@ void pack_a(hipStream_t stream, const argument& result, const argument& arg) ...@@ -40,11 +40,11 @@ void pack_a(hipStream_t stream, const argument& result, const argument& arg)
void pack_b(hipStream_t stream, const argument& result, const argument& arg) void pack_b(hipStream_t stream, const argument& result, const argument& arg)
{ {
auto trans_shape = result.get_shape(); auto trans_shape = result.get_shape();
auto out_lens = trans_shape.lens(); auto out_lens = trans_shape.lens();
auto dim_0 = trans_shape.lens().size() - 2; auto dim_0 = trans_shape.lens().size() - 2;
auto dim_1 = trans_shape.lens().size() - 1; auto dim_1 = trans_shape.lens().size() - 1;
std::size_t ldb = trans_shape.strides()[dim_1]; std::size_t ldb = trans_shape.strides()[dim_1];
auto wrap_lens = out_lens; auto wrap_lens = out_lens;
std::swap(wrap_lens[dim_0], wrap_lens[dim_1]); std::swap(wrap_lens[dim_0], wrap_lens[dim_1]);
......
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