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>
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)
{
......
......@@ -59,7 +59,7 @@ constexpr auto gemm_batch_slice(Tensor t, T i)
{
constexpr auto batch = gemm_get_batches<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());
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