Commit d8110fc4 authored by Paul's avatar Paul
Browse files

Format

parent acb80f40
...@@ -59,7 +59,8 @@ constexpr auto gemm_batch_slice(Tensor t, T i) ...@@ -59,7 +59,8 @@ constexpr auto gemm_batch_slice(Tensor t, T i)
{ {
constexpr auto batch = gemm_get_batches<Tensor>(); constexpr auto batch = gemm_get_batches<Tensor>();
constexpr auto matrix = gemm_get_matrix<Tensor>(); constexpr auto matrix = gemm_get_matrix<Tensor>();
MIGRAPHX_ASSERT((batch.index(i) + matrix.get_shape().element_space()) <= t.get_shape().element_space()); MIGRAPHX_ASSERT((batch.index(i) + matrix.get_shape().element_space()) <=
t.get_shape().element_space());
return make_tensor_view(t.data() + batch.index(i), matrix); return make_tensor_view(t.data() + batch.index(i), matrix);
} }
......
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