adds kUnknown type, sizes half properly

Author: MichealReed

gpu.hpp

@@ -204,6 +204,7 @@ enum NumType {
   ku16,
   ku32,
   ku64,
+  kUnknown
 };
 
 /**
@@ -212,15 +213,15 @@ enum NumType {
 inline size_t sizeBytes(const NumType &type) {
   switch (type) {
   case kf16:
-    return sizeof(uint16_t);
+    return sizeof(half);
   case kf32:
     return sizeof(float);
   case kf64:
     return sizeof(double);
   case ki8:
-    return sizeof(uint8_t);
+    return sizeof(int8_t);
   case ki16:
-    return sizeof(uint16_t);
+    return sizeof(int16_t);
   case ki32:
     return sizeof(int32_t);
   case ki64:

test/test_gpu.cpp

@@ -218,8 +218,7 @@ void testAddKernelInt8() {
   Kernel unpackKernelB =
       createKernel(ctx, {kPackedInt8ToInt32Kernel, workgroupSize, ki32},
                    Bindings{bTensorPacked, bTensorUnpacked},
-                   {cdiv(packedCount, workgroupSize), 1,
-                    1}); 
+                   {cdiv(packedCount, workgroupSize), 1, 1});
   // Dispatch based on packed size
   dispatchKernel(ctx, unpackKernelA);
   dispatchKernel(ctx, unpackKernelB);
@@ -245,8 +244,6 @@ void testAddKernelInt8() {
   toCPU(ctx, outputTensorPacked, ki8, result.data(), 0);
 
   for (size_t i = 0; i < N; ++i) {
-    LOG(kDefLog, kInfo, "result[%zu] = %d, expected[%zu] = %d", i, result[i], i,
-        expected[i]);
     assert(result[i] == expected[i]);
   }