add fp8-fbgemm kernels (#86)

* add fp8-fbgemm kernels

Signed-off-by: Wang, Yi A <[email protected]>

* add license

Signed-off-by: Wang, Yi A <[email protected]>

---------

Signed-off-by: Wang, Yi A <[email protected]>

Author: sywangyi

fp8-fbgemm/build.toml

@@ -0,0 +1,3 @@
+[general]
+name = "fp8-fbgemm"
+universal = true

fp8-fbgemm/flake.lock

@@ -0,0 +1,17 @@
+{
+  description = "Flake for fp8-fbgemm kernels";
+
+  inputs = {
+    kernel-builder.url = "github:huggingface/kernel-builder";
+  };
+
+  outputs =
+    {
+      self,
+      kernel-builder,
+    }:
+    kernel-builder.lib.genFlakeOutputs {
+      inherit self;
+      path = ./.;
+    };
+}

fp8-fbgemm/flake.nix

@@ -0,0 +1,5 @@
+from .quantizer import quantize_fp8_per_row
+
+__all__ = [
+    quantize_fp8_per_row
+]