implement private hbm only

KLR/Core/Tensor.lean

@@ -29,7 +29,7 @@ open Util (FromSexp ToSexp)
 -- Memory types on hardware
 @[serde tag = 110]
 inductive Memory where
-  | hbm | sbuf | psum | reg
+  | hbm | sbuf | psum | reg | private_hbm
   deriving BEq, FromCBOR, FromJson, FromSexp, Repr, ToCBOR, ToJson, ToSexp
 
 namespace Memory
@@ -39,6 +39,7 @@ def toName (m : Memory) :=
   | .sbuf => `sbuf
   | .psum => `psum
   | .reg => `reg
+  | .private_hbm => `private_hbm
 end Memory
 
 -- register number

interop/klr/NKI.asdl

@@ -9,6 +9,7 @@ Memory =
   | sbuf
   | psum
   | reg
+  | private_hbm
 Dtype =
   | bfloat16
   | float8e3

interop/klr/klir_ast.hpp

@@ -58,6 +58,7 @@ enum class Memory {
   sbuf,
   psum,
   reg,
+  private_hbm,
 };
 
 enum class Dtype {

interop/klr/klir_pretty_print.cpp

@@ -121,6 +121,9 @@ std::string to_string(Memory &MemoryInstance) {
   case Memory::reg:
     return "reg";
     break;
+  case Memory::private_hbm:
+    return "private_hbm";
+    break;
   default:
     return "UNABLE TO PRINT";
   }

interop/klr/klir_serde.cpp

@@ -403,6 +403,12 @@ Memory Memory_des(FILE *in) {
     return Memory::reg;
     break;
   }
+  case 4: {
+    if (l != 0)
+      throw std::runtime_error("Wrong number of elements");
+    return Memory::private_hbm;
+    break;
+  }
   default:
     throw std::runtime_error("Invalid value tag");
   }