Commit c878fa25 authored by Alan Turner's avatar Alan Turner
Browse files

Remove redundant get_shape and remove ck_gemm from names

parent aa4343ac
...@@ -265,7 +265,7 @@ struct ck_gemm_compiler : compiler<ck_gemm_compiler> ...@@ -265,7 +265,7 @@ struct ck_gemm_compiler : compiler<ck_gemm_compiler>
s = shape{s.type(), {m1, m2}}; s = shape{s.type(), {m1, m2}};
} }
std::vector<std::string> names() const { return {"ck_gemm", "gpu::ck_gemm"}; } std::vector<std::string> names() const { return {"gpu::ck_gemm"}; }
static bool standard_batch(const shape& s) static bool standard_batch(const shape& s)
{ {
......
...@@ -59,7 +59,7 @@ constexpr auto gemm_batch_slice(Tensor t, T i) ...@@ -59,7 +59,7 @@ 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()) <= MIGRAPHX_ASSERT((batch.index(i) + matrix.element_space()) <=
t.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