Upload folder using huggingface_hub

.gitattributes

@@ -33,3 +33,12 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+build/torch27-cxx11-cu118-x86_64-linux/yoso/_yoso_3173620_dirty.abi3.so filter=lfs diff=lfs merge=lfs -text
+build/torch27-cxx11-cu126-x86_64-linux/yoso/_yoso_3173620_dirty.abi3.so filter=lfs diff=lfs merge=lfs -text
+build/torch27-cxx11-cu128-x86_64-linux/yoso/_yoso_3173620_dirty.abi3.so filter=lfs diff=lfs merge=lfs -text
+build/torch28-cxx11-cu126-x86_64-linux/yoso/_yoso_3173620_dirty.abi3.so filter=lfs diff=lfs merge=lfs -text
+build/torch28-cxx11-cu128-x86_64-linux/yoso/_yoso_3173620_dirty.abi3.so filter=lfs diff=lfs merge=lfs -text
+build/torch28-cxx11-cu129-x86_64-linux/yoso/_yoso_3173620_dirty.abi3.so filter=lfs diff=lfs merge=lfs -text
+build/torch29-cxx11-cu126-x86_64-linux/yoso/_yoso_3173620_dirty.abi3.so filter=lfs diff=lfs merge=lfs -text
+build/torch29-cxx11-cu128-x86_64-linux/yoso/_yoso_3173620_dirty.abi3.so filter=lfs diff=lfs merge=lfs -text
+build/torch29-cxx11-cu130-x86_64-linux/yoso/_yoso_3173620_dirty.abi3.so filter=lfs diff=lfs merge=lfs -text

build.toml

@@ -0,0 +1,32 @@
+[general]
+name = "yoso"
+universal = false
+
+[torch]
+src = [
+  "torch-ext/torch_binding.cpp",
+  "torch-ext/torch_binding.h",
+]
+
+[kernel.yoso]
+depends = ["torch"]
+backend = "cuda"
+cuda-capabilities = [
+    "8.0",
+    "8.9",
+    "9.0",
+    "10.0",
+    "12.0",
+]
+include = ["."]
+src = [
+    "yoso/fast_lsh_cumulation_cuda.cu",
+    "yoso/fast_lsh_cumulation_cuda.h",
+    "yoso/fast_lsh_cumulation_torch.cpp",
+    "yoso/fast_lsh_cumulation.h",
+    "yoso/fast_lsh_cumulation.cu",
+    "yoso/common_cuda.h",
+    "yoso/common.h",
+    "yoso/common_cuda_device.h",
+]
+

build/torch27-cxx11-cu118-x86_64-linux/yoso/__init__.py

@@ -0,0 +1,11 @@
+from ._ops import ops
+
+fast_hash = ops.fast_hash
+lsh_cumulation = ops.lsh_cumulation
+lsh_weighted_cumulation = ops.lsh_weighted_cumulation
+
+__all__ = [
+    "fast_hash",
+    "lsh_cumulation",
+    "lsh_weighted_cumulation",
+]

build/torch27-cxx11-cu118-x86_64-linux/yoso/_ops.py

@@ -0,0 +1,9 @@
+import torch
+from . import _yoso_3173620_dirty
+ops = torch.ops._yoso_3173620_dirty
+
+def add_op_namespace_prefix(op_name: str):
+    """
+    Prefix op by namespace.
+    """
+    return f"_yoso_3173620_dirty::{op_name}"

build/torch27-cxx11-cu118-x86_64-linux/yoso/_yoso_3173620_dirty.abi3.so

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a993d93b5c46e4056a29a9c8a72b5499f01d6c852f6a8ee3662716dcf0bfe08a
+size 2465968

build/torch27-cxx11-cu126-x86_64-linux/yoso/__init__.py

@@ -0,0 +1,11 @@
+from ._ops import ops
+
+fast_hash = ops.fast_hash
+lsh_cumulation = ops.lsh_cumulation
+lsh_weighted_cumulation = ops.lsh_weighted_cumulation
+
+__all__ = [
+    "fast_hash",
+    "lsh_cumulation",
+    "lsh_weighted_cumulation",
+]

build/torch27-cxx11-cu126-x86_64-linux/yoso/_ops.py

@@ -0,0 +1,9 @@
+import torch
+from . import _yoso_3173620_dirty
+ops = torch.ops._yoso_3173620_dirty
+
+def add_op_namespace_prefix(op_name: str):
+    """
+    Prefix op by namespace.
+    """
+    return f"_yoso_3173620_dirty::{op_name}"

build/torch27-cxx11-cu126-x86_64-linux/yoso/_yoso_3173620_dirty.abi3.so

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:557df99aef151fadf862a138f84943dd93cb898211a527787668f191e63dc107
+size 2494960

build/torch27-cxx11-cu128-x86_64-linux/yoso/__init__.py

@@ -0,0 +1,11 @@
+from ._ops import ops
+
+fast_hash = ops.fast_hash
+lsh_cumulation = ops.lsh_cumulation
+lsh_weighted_cumulation = ops.lsh_weighted_cumulation
+
+__all__ = [
+    "fast_hash",
+    "lsh_cumulation",
+    "lsh_weighted_cumulation",
+]

build/torch27-cxx11-cu128-x86_64-linux/yoso/_ops.py

@@ -0,0 +1,9 @@
+import torch
+from . import _yoso_3173620_dirty
+ops = torch.ops._yoso_3173620_dirty
+
+def add_op_namespace_prefix(op_name: str):
+    """
+    Prefix op by namespace.
+    """
+    return f"_yoso_3173620_dirty::{op_name}"

build/torch27-cxx11-cu128-x86_64-linux/yoso/_yoso_3173620_dirty.abi3.so

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d316cd26d90f963527aa7a68e2340545635c2201bd69c781a1040fb5ea6aa54b
+size 3045736

build/torch28-cxx11-cu126-x86_64-linux/yoso/__init__.py

@@ -0,0 +1,11 @@
+from ._ops import ops
+
+fast_hash = ops.fast_hash
+lsh_cumulation = ops.lsh_cumulation
+lsh_weighted_cumulation = ops.lsh_weighted_cumulation
+
+__all__ = [
+    "fast_hash",
+    "lsh_cumulation",
+    "lsh_weighted_cumulation",
+]

build/torch28-cxx11-cu126-x86_64-linux/yoso/_ops.py

@@ -0,0 +1,9 @@
+import torch
+from . import _yoso_3173620_dirty
+ops = torch.ops._yoso_3173620_dirty
+
+def add_op_namespace_prefix(op_name: str):
+    """
+    Prefix op by namespace.
+    """
+    return f"_yoso_3173620_dirty::{op_name}"

build/torch28-cxx11-cu126-x86_64-linux/yoso/_yoso_3173620_dirty.abi3.so

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b232acaeb039ea7c2476dcf1d4618f90ff5d8af726126ed7d80deef4d8195cef
+size 2499304

build/torch28-cxx11-cu128-x86_64-linux/yoso/__init__.py

@@ -0,0 +1,11 @@
+from ._ops import ops
+
+fast_hash = ops.fast_hash
+lsh_cumulation = ops.lsh_cumulation
+lsh_weighted_cumulation = ops.lsh_weighted_cumulation
+
+__all__ = [
+    "fast_hash",
+    "lsh_cumulation",
+    "lsh_weighted_cumulation",
+]

build/torch28-cxx11-cu128-x86_64-linux/yoso/_ops.py

@@ -0,0 +1,9 @@
+import torch
+from . import _yoso_3173620_dirty
+ops = torch.ops._yoso_3173620_dirty
+
+def add_op_namespace_prefix(op_name: str):
+    """
+    Prefix op by namespace.
+    """
+    return f"_yoso_3173620_dirty::{op_name}"

build/torch28-cxx11-cu128-x86_64-linux/yoso/_yoso_3173620_dirty.abi3.so

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bd27666f9ad974f958ffeb0279e54973aa4b7fa26a94ca855371ebd54de57ece
+size 3045960

build/torch28-cxx11-cu129-x86_64-linux/yoso/__init__.py

@@ -0,0 +1,11 @@
+from ._ops import ops
+
+fast_hash = ops.fast_hash
+lsh_cumulation = ops.lsh_cumulation
+lsh_weighted_cumulation = ops.lsh_weighted_cumulation
+
+__all__ = [
+    "fast_hash",
+    "lsh_cumulation",
+    "lsh_weighted_cumulation",
+]

build/torch28-cxx11-cu129-x86_64-linux/yoso/_ops.py

@@ -0,0 +1,9 @@
+import torch
+from . import _yoso_3173620_dirty
+ops = torch.ops._yoso_3173620_dirty
+
+def add_op_namespace_prefix(op_name: str):
+    """
+    Prefix op by namespace.
+    """
+    return f"_yoso_3173620_dirty::{op_name}"

build/torch28-cxx11-cu129-x86_64-linux/yoso/_yoso_3173620_dirty.abi3.so

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:67cfb08278e43598e473113ac2a67522b922a0859d008e18ad5137a112ea4df7
+size 3055168

build/torch29-cxx11-cu126-x86_64-linux/yoso/__init__.py

@@ -0,0 +1,11 @@
+from ._ops import ops
+
+fast_hash = ops.fast_hash
+lsh_cumulation = ops.lsh_cumulation
+lsh_weighted_cumulation = ops.lsh_weighted_cumulation
+
+__all__ = [
+    "fast_hash",
+    "lsh_cumulation",
+    "lsh_weighted_cumulation",
+]

build/torch29-cxx11-cu126-x86_64-linux/yoso/_ops.py

@@ -0,0 +1,9 @@
+import torch
+from . import _yoso_3173620_dirty
+ops = torch.ops._yoso_3173620_dirty
+
+def add_op_namespace_prefix(op_name: str):
+    """
+    Prefix op by namespace.
+    """
+    return f"_yoso_3173620_dirty::{op_name}"

build/torch29-cxx11-cu126-x86_64-linux/yoso/_yoso_3173620_dirty.abi3.so

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:59b2e7ead396d2c977cc0523a5f5f112d9fbc85b637cca5e4fd07c1062accbdf
+size 2499280

build/torch29-cxx11-cu128-x86_64-linux/yoso/__init__.py

@@ -0,0 +1,11 @@
+from ._ops import ops
+
+fast_hash = ops.fast_hash
+lsh_cumulation = ops.lsh_cumulation
+lsh_weighted_cumulation = ops.lsh_weighted_cumulation
+
+__all__ = [
+    "fast_hash",
+    "lsh_cumulation",
+    "lsh_weighted_cumulation",
+]

build/torch29-cxx11-cu128-x86_64-linux/yoso/_ops.py

@@ -0,0 +1,9 @@
+import torch
+from . import _yoso_3173620_dirty
+ops = torch.ops._yoso_3173620_dirty
+
+def add_op_namespace_prefix(op_name: str):
+    """
+    Prefix op by namespace.
+    """
+    return f"_yoso_3173620_dirty::{op_name}"

build/torch29-cxx11-cu128-x86_64-linux/yoso/_yoso_3173620_dirty.abi3.so

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d1079a87dca1bb50c85509094efa36b42a717375a2a2e795ae917c1b47f51b43
+size 3045936

build/torch29-cxx11-cu130-x86_64-linux/yoso/__init__.py

@@ -0,0 +1,11 @@
+from ._ops import ops
+
+fast_hash = ops.fast_hash
+lsh_cumulation = ops.lsh_cumulation
+lsh_weighted_cumulation = ops.lsh_weighted_cumulation
+
+__all__ = [
+    "fast_hash",
+    "lsh_cumulation",
+    "lsh_weighted_cumulation",
+]

build/torch29-cxx11-cu130-x86_64-linux/yoso/_ops.py

@@ -0,0 +1,9 @@
+import torch
+from . import _yoso_3173620_dirty
+ops = torch.ops._yoso_3173620_dirty
+
+def add_op_namespace_prefix(op_name: str):
+    """
+    Prefix op by namespace.
+    """
+    return f"_yoso_3173620_dirty::{op_name}"

build/torch29-cxx11-cu130-x86_64-linux/yoso/_yoso_3173620_dirty.abi3.so

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:298f1c5374518a1d6497e22171492ce0113a6b367cb40beb20782005b93a9745
+size 3099592

flake.lock

@@ -0,0 +1,168 @@
+{
+  "nodes": {
+    "flake-compat": {
+      "locked": {
+        "lastModified": 1747046372,
+        "narHash": "sha256-CIVLLkVgvHYbgI2UpXvIIBJ12HWgX+fjA8Xf8PUmqCY=",
+        "owner": "edolstra",
+        "repo": "flake-compat",
+        "rev": "9100a0f413b0c601e0533d1d94ffd501ce2e7885",
+        "type": "github"
+      },
+      "original": {
+        "owner": "edolstra",
+        "repo": "flake-compat",
+        "type": "github"
+      }
+    },
+    "flake-compat_2": {
+      "locked": {
+        "lastModified": 1747046372,
+        "narHash": "sha256-CIVLLkVgvHYbgI2UpXvIIBJ12HWgX+fjA8Xf8PUmqCY=",
+        "owner": "edolstra",
+        "repo": "flake-compat",
+        "rev": "9100a0f413b0c601e0533d1d94ffd501ce2e7885",
+        "type": "github"
+      },
+      "original": {
+        "owner": "edolstra",
+        "repo": "flake-compat",
+        "type": "github"
+      }
+    },
+    "flake-utils": {
+      "inputs": {
+        "systems": "systems"
+      },
+      "locked": {
+        "lastModified": 1731533236,
+        "narHash": "sha256-l0KFg5HjrsfsO/JpG+r7fRrqm12kzFHyUHqHCVpMMbI=",
+        "owner": "numtide",
+        "repo": "flake-utils",
+        "rev": "11707dc2f618dd54ca8739b309ec4fc024de578b",
+        "type": "github"
+      },
+      "original": {
+        "owner": "numtide",
+        "repo": "flake-utils",
+        "type": "github"
+      }
+    },
+    "flake-utils_2": {
+      "inputs": {
+        "systems": "systems_2"
+      },
+      "locked": {
+        "lastModified": 1731533236,
+        "narHash": "sha256-l0KFg5HjrsfsO/JpG+r7fRrqm12kzFHyUHqHCVpMMbI=",
+        "owner": "numtide",
+        "repo": "flake-utils",
+        "rev": "11707dc2f618dd54ca8739b309ec4fc024de578b",
+        "type": "github"
+      },
+      "original": {
+        "owner": "numtide",
+        "repo": "flake-utils",
+        "type": "github"
+      }
+    },
+    "hf-nix": {
+      "inputs": {
+        "flake-compat": "flake-compat_2",
+        "flake-utils": "flake-utils_2",
+        "nixpkgs": "nixpkgs"
+      },
+      "locked": {
+        "lastModified": 1759851564,
+        "narHash": "sha256-Xybkhm0FM/VzlZ5WndTYq/X/9MAeddd4EQ2Vz8GdkOA=",
+        "owner": "huggingface",
+        "repo": "hf-nix",
+        "rev": "351655d9f124805ed7c1193aa61550ce245f4570",
+        "type": "github"
+      },
+      "original": {
+        "owner": "huggingface",
+        "repo": "hf-nix",
+        "type": "github"
+      }
+    },
+    "kernel-builder": {
+      "inputs": {
+        "flake-compat": "flake-compat",
+        "flake-utils": "flake-utils",
+        "hf-nix": "hf-nix",
+        "nixpkgs": [
+          "kernel-builder",
+          "hf-nix",
+          "nixpkgs"
+        ]
+      },
+      "locked": {
+        "lastModified": 1760035358,
+        "narHash": "sha256-N5vmCrgwcIluPclf/hmnofLK77EJJYh5PR8SRvw++es=",
+        "owner": "huggingface",
+        "repo": "kernel-builder",
+        "rev": "a48cbd19ae7e425dfc1865188ef06dac43ab9244",
+        "type": "github"
+      },
+      "original": {
+        "owner": "huggingface",
+        "repo": "kernel-builder",
+        "type": "github"
+      }
+    },
+    "nixpkgs": {
+      "locked": {
+        "lastModified": 1755963616,
+        "narHash": "sha256-6yD0ww/S8n+U2uPYcJZ3DRURP8Kx036GRpR2uPNZroE=",
+        "owner": "nixos",
+        "repo": "nixpkgs",
+        "rev": "73e96df7cff5783f45e21342a75a1540c4eddce4",
+        "type": "github"
+      },
+      "original": {
+        "owner": "nixos",
+        "ref": "nixos-unstable-small",
+        "repo": "nixpkgs",
+        "type": "github"
+      }
+    },
+    "root": {
+      "inputs": {
+        "kernel-builder": "kernel-builder"
+      }
+    },
+    "systems": {
+      "locked": {
+        "lastModified": 1681028828,
+        "narHash": "sha256-Vy1rq5AaRuLzOxct8nz4T6wlgyUR7zLU309k9mBC768=",
+        "owner": "nix-systems",
+        "repo": "default",
+        "rev": "da67096a3b9bf56a91d16901293e51ba5b49a27e",
+        "type": "github"
+      },
+      "original": {
+        "owner": "nix-systems",
+        "repo": "default",
+        "type": "github"
+      }
+    },
+    "systems_2": {
+      "locked": {
+        "lastModified": 1681028828,
+        "narHash": "sha256-Vy1rq5AaRuLzOxct8nz4T6wlgyUR7zLU309k9mBC768=",
+        "owner": "nix-systems",
+        "repo": "default",
+        "rev": "da67096a3b9bf56a91d16901293e51ba5b49a27e",
+        "type": "github"
+      },
+      "original": {
+        "owner": "nix-systems",
+        "repo": "default",
+        "type": "github"
+      }
+    }
+  },
+  "root": "root",
+  "version": 7
+}

flake.nix

@@ -0,0 +1,17 @@
+{
+  description = "Flake for yoso kernels";
+
+  inputs = {
+    kernel-builder.url = "github:huggingface/kernel-builder/";
+  };
+
+  outputs =
+    {
+      self,
+      kernel-builder,
+    }:
+    kernel-builder.lib.genFlakeOutputs {
+      path = ./.;
+      rev = self.shortRev or self.dirtyShortRev or self.lastModifiedDate;
+    };
+}