danieldk HF Staff commited on
Commit
7890526
·
verified ·
1 Parent(s): 09b1f1e

Build uploaded using `kernels`.

Browse files
.gitattributes CHANGED
@@ -54,3 +54,4 @@ build/torch210-cxx11-cu130-x86_64-linux/_cv_utils_cuda_8cfc281.abi3.so filter=lf
54
  build/torch29-cxx11-cu126-x86_64-linux/_cv_utils_cuda_8cfc281.abi3.so filter=lfs diff=lfs merge=lfs -text
55
  build/torch29-cxx11-cu128-x86_64-linux/_cv_utils_cuda_8cfc281.abi3.so filter=lfs diff=lfs merge=lfs -text
56
  build/torch29-cxx11-cu130-x86_64-linux/_cv_utils_cuda_8cfc281.abi3.so filter=lfs diff=lfs merge=lfs -text
 
 
54
  build/torch29-cxx11-cu126-x86_64-linux/_cv_utils_cuda_8cfc281.abi3.so filter=lfs diff=lfs merge=lfs -text
55
  build/torch29-cxx11-cu128-x86_64-linux/_cv_utils_cuda_8cfc281.abi3.so filter=lfs diff=lfs merge=lfs -text
56
  build/torch29-cxx11-cu130-x86_64-linux/_cv_utils_cuda_8cfc281.abi3.so filter=lfs diff=lfs merge=lfs -text
57
+ build/torch210-cu128-x86_64-windows/_cv_utils_cuda_0561e14.pyd filter=lfs diff=lfs merge=lfs -text
build/torch210-cu128-x86_64-windows/__init__.py ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+ from typing import List
3
+
4
+ from ._ops import ops
5
+
6
+ def cc_2d(inputs: torch.Tensor, get_counts: bool) -> List[torch.Tensor]:
7
+ return ops.cc_2d(inputs, get_counts)
8
+
9
+ def generic_nms(dets: torch.Tensor, scores: torch.Tensor, iou_threshold: float, use_iou_matrix: bool) -> torch.Tensor:
10
+ return ops.generic_nms(dets, scores, iou_threshold, use_iou_matrix)
11
+
12
+ __all__ = ["cc_2d", "generic_nms"]
build/torch210-cu128-x86_64-windows/_cv_utils_cuda_0561e14.pyd ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:77031aa43cd474f524a591413a7c46c35a4f05a625df3ddf0b7b12d69029da50
3
+ size 1782784
build/torch210-cu128-x86_64-windows/_ops.py ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+ from . import _cv_utils_cuda_0561e14
3
+ ops = torch.ops._cv_utils_cuda_0561e14
4
+
5
+ def add_op_namespace_prefix(op_name: str):
6
+ """
7
+ Prefix op by namespace.
8
+ """
9
+ return f"_cv_utils_cuda_0561e14::{op_name}"
build/torch210-cu128-x86_64-windows/cv_utils/__init__.py ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import ctypes
2
+ import sys
3
+
4
+ import importlib
5
+ from pathlib import Path
6
+ from types import ModuleType
7
+
8
+ def _import_from_path(file_path: Path) -> ModuleType:
9
+ # We cannot use the module name as-is, after adding it to `sys.modules`,
10
+ # it would also be used for other imports. So, we make a module name that
11
+ # depends on the path for it to be unique using the hex-encoded hash of
12
+ # the path.
13
+ path_hash = "{:x}".format(ctypes.c_size_t(hash(file_path.absolute())).value)
14
+ module_name = path_hash
15
+ spec = importlib.util.spec_from_file_location(module_name, file_path)
16
+ if spec is None:
17
+ raise ImportError(f"Cannot load spec for {module_name} from {file_path}")
18
+ module = importlib.util.module_from_spec(spec)
19
+ if module is None:
20
+ raise ImportError(f"Cannot load module {module_name} from spec")
21
+ sys.modules[module_name] = module
22
+ spec.loader.exec_module(module) # type: ignore
23
+ return module
24
+
25
+
26
+ globals().update(vars(_import_from_path(Path(__file__).parent.parent / "__init__.py")))
build/torch210-cu128-x86_64-windows/metadata.json ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "version": 1,
3
+ "python-depends": [],
4
+ "backend": {
5
+ "type": "cuda",
6
+ "archs": [
7
+ "10.0",
8
+ "10.1",
9
+ "12.0+PTX",
10
+ "7.0",
11
+ "7.2",
12
+ "7.5",
13
+ "8.0",
14
+ "8.6",
15
+ "8.7",
16
+ "8.9",
17
+ "9.0"
18
+ ]
19
+ }
20
+ }