Add assertion and missing headers

src/targets/gpu/jit/ck_gemm.cpp

@@ -58,6 +58,7 @@ static const char* const ck_gemm_kernel = R"__migraphx__(
 #include <args.hpp>
 #include <migraphx/kernels/ck_gemm.hpp>
 #include <migraphx/kernels/pointwise.hpp>
+#include <migraphx/kernels/ops.hpp>
 
 namespace migraphx {
 

src/targets/gpu/kernels/include/migraphx/kernels/ck_gemm_includes.hpp

@@ -29,6 +29,7 @@
 #include <migraphx/kernels/integral_constant.hpp>
 #include <migraphx/kernels/tensor_view.hpp>
 
+#include <iostream>
 #include <ck/utility/common_header.hpp>
 #include <ck/tensor_description/tensor_descriptor.hpp>
 #include <ck/tensor_description/tensor_descriptor_helper.hpp>

src/targets/gpu/kernels/include/migraphx/kernels/gemm_batcher.hpp

@@ -36,6 +36,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()) <= t.get_shape().element_space());
     return make_tensor_view(t.data() + batch.index(i), matrix);
 }