re-apply diff

Author: tarang-jain

cpp/include/raft/linalg/contractions.cuh

@@ -159,6 +159,12 @@ struct Policy4x4<half, _veclen> {
   typedef KernelPolicy<half, _veclen, 64, 4, 4, 16, 16> Policy;
   typedef ColKernelPolicy<half, _veclen, 64, 4, 4, 16, 16> ColPolicy;
 };
+
+template <int _veclen>
+struct Policy4x4<uint8_t, _veclen> {
+  typedef KernelPolicy<uint8_t, _veclen, 64, 4, 4, 16, 16> Policy;
+  typedef ColKernelPolicy<uint8_t, _veclen, 64, 4, 4, 16, 16> ColPolicy;
+};
 /** @} */
 
 /**
@@ -182,6 +188,12 @@ struct Policy4x4Skinny<double, _veclen> {
   typedef ColKernelPolicy<double, _veclen, 8, 4, 4, 8, 8> ColPolicy;
 };
 
+template <int _veclen>
+struct Policy4x4Skinny<uint8_t, _veclen> {
+  typedef KernelPolicy<uint8_t, _veclen, 8, 4, 4, 8, 8> Policy;
+  typedef ColKernelPolicy<uint8_t, _veclen, 8, 4, 4, 8, 8> ColPolicy;
+};
+
 /**
  * @defgroup Policy2x8 16 elements per thread Policy with k-block = 16
  * @{