Commit
·
6a2f5d2
1
Parent(s):
872b645
rename cv_utils
Browse files- .gitattributes +6 -6
- README.md +2 -2
- build.toml +4 -4
- build/torch28-cxx11-cu126-x86_64-linux/sam3_kernels/_ops.py +4 -3
- build/torch28-cxx11-cu128-x86_64-linux/sam3_kernels/_ops.py +4 -3
- build/torch28-cxx11-cu129-x86_64-linux/sam3_kernels/_ops.py +4 -3
- build/torch29-cxx11-cu126-x86_64-linux/sam3_kernels/_ops.py +4 -3
- build/torch29-cxx11-cu128-x86_64-linux/sam3_kernels/_ops.py +4 -3
- build/torch29-cxx11-cu130-x86_64-linux/sam3_kernels/_ops.py +4 -3
.gitattributes
CHANGED
|
@@ -33,9 +33,9 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
-
build/torch28-cxx11-cu126-x86_64-linux/
|
| 37 |
-
build/torch28-cxx11-cu128-x86_64-linux/
|
| 38 |
-
build/torch28-cxx11-cu129-x86_64-linux/
|
| 39 |
-
build/torch29-cxx11-cu126-x86_64-linux/
|
| 40 |
-
build/torch29-cxx11-cu128-x86_64-linux/
|
| 41 |
-
build/torch29-cxx11-cu130-x86_64-linux/
|
|
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
+
build/torch28-cxx11-cu126-x86_64-linux/cv_utils/_cv_utils_19700101000000.abi3.so filter=lfs diff=lfs merge=lfs -text
|
| 37 |
+
build/torch28-cxx11-cu128-x86_64-linux/cv_utils/_cv_utils_19700101000000.abi3.so filter=lfs diff=lfs merge=lfs -text
|
| 38 |
+
build/torch28-cxx11-cu129-x86_64-linux/cv_utils/_cv_utils_19700101000000.abi3.so filter=lfs diff=lfs merge=lfs -text
|
| 39 |
+
build/torch29-cxx11-cu126-x86_64-linux/cv_utils/_cv_utils_19700101000000.abi3.so filter=lfs diff=lfs merge=lfs -text
|
| 40 |
+
build/torch29-cxx11-cu128-x86_64-linux/cv_utils/_cv_utils_19700101000000.abi3.so filter=lfs diff=lfs merge=lfs -text
|
| 41 |
+
build/torch29-cxx11-cu130-x86_64-linux/cv_utils/_cv_utils_19700101000000.abi3.so filter=lfs diff=lfs merge=lfs -text
|
README.md
CHANGED
|
@@ -3,6 +3,6 @@ tags:
|
|
| 3 |
- kernels
|
| 4 |
---
|
| 5 |
|
| 6 |
-
#
|
| 7 |
|
| 8 |
-
This is a build for some kernel utilities for computer vision, including generic_nms and connected_components
|
|
|
|
| 3 |
- kernels
|
| 4 |
---
|
| 5 |
|
| 6 |
+
# cv_utils
|
| 7 |
|
| 8 |
+
This is a build for some kernel utilities for computer vision, including generic_nms and connected_components
|
build.toml
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
[general]
|
| 2 |
-
name = "
|
| 3 |
universal = false
|
| 4 |
|
| 5 |
[torch]
|
|
@@ -8,13 +8,13 @@ src = [
|
|
| 8 |
"torch-ext/torch_binding.h",
|
| 9 |
]
|
| 10 |
|
| 11 |
-
[kernel.
|
| 12 |
depends = ["torch"]
|
| 13 |
backend = "cuda"
|
| 14 |
|
| 15 |
src = [
|
| 16 |
-
"
|
| 17 |
-
"
|
| 18 |
]
|
| 19 |
|
| 20 |
cuda-flags = [
|
|
|
|
| 1 |
[general]
|
| 2 |
+
name = "cv_utils"
|
| 3 |
universal = false
|
| 4 |
|
| 5 |
[torch]
|
|
|
|
| 8 |
"torch-ext/torch_binding.h",
|
| 9 |
]
|
| 10 |
|
| 11 |
+
[kernel.cv_utils]
|
| 12 |
depends = ["torch"]
|
| 13 |
backend = "cuda"
|
| 14 |
|
| 15 |
src = [
|
| 16 |
+
"cv_utils/connected_components.cu",
|
| 17 |
+
"cv_utils/generic_nms.cu",
|
| 18 |
]
|
| 19 |
|
| 20 |
cuda-flags = [
|
build/torch28-cxx11-cu126-x86_64-linux/sam3_kernels/_ops.py
CHANGED
|
@@ -1,9 +1,10 @@
|
|
| 1 |
import torch
|
| 2 |
-
|
| 3 |
-
ops = torch.ops.
|
|
|
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
-
return f"
|
|
|
|
| 1 |
import torch
|
| 2 |
+
|
| 3 |
+
ops = torch.ops._cv_utils_19700101000000
|
| 4 |
+
|
| 5 |
|
| 6 |
def add_op_namespace_prefix(op_name: str):
|
| 7 |
"""
|
| 8 |
Prefix op by namespace.
|
| 9 |
"""
|
| 10 |
+
return f"_cv_utils_19700101000000::{op_name}"
|
build/torch28-cxx11-cu128-x86_64-linux/sam3_kernels/_ops.py
CHANGED
|
@@ -1,9 +1,10 @@
|
|
| 1 |
import torch
|
| 2 |
-
|
| 3 |
-
ops = torch.ops.
|
|
|
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
-
return f"
|
|
|
|
| 1 |
import torch
|
| 2 |
+
|
| 3 |
+
ops = torch.ops._cv_utils_19700101000000
|
| 4 |
+
|
| 5 |
|
| 6 |
def add_op_namespace_prefix(op_name: str):
|
| 7 |
"""
|
| 8 |
Prefix op by namespace.
|
| 9 |
"""
|
| 10 |
+
return f"_cv_utils_19700101000000::{op_name}"
|
build/torch28-cxx11-cu129-x86_64-linux/sam3_kernels/_ops.py
CHANGED
|
@@ -1,9 +1,10 @@
|
|
| 1 |
import torch
|
| 2 |
-
|
| 3 |
-
ops = torch.ops.
|
|
|
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
-
return f"
|
|
|
|
| 1 |
import torch
|
| 2 |
+
|
| 3 |
+
ops = torch.ops._cv_utils_19700101000000
|
| 4 |
+
|
| 5 |
|
| 6 |
def add_op_namespace_prefix(op_name: str):
|
| 7 |
"""
|
| 8 |
Prefix op by namespace.
|
| 9 |
"""
|
| 10 |
+
return f"_cv_utils_19700101000000::{op_name}"
|
build/torch29-cxx11-cu126-x86_64-linux/sam3_kernels/_ops.py
CHANGED
|
@@ -1,9 +1,10 @@
|
|
| 1 |
import torch
|
| 2 |
-
|
| 3 |
-
ops = torch.ops.
|
|
|
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
-
return f"
|
|
|
|
| 1 |
import torch
|
| 2 |
+
|
| 3 |
+
ops = torch.ops._cv_utils_19700101000000
|
| 4 |
+
|
| 5 |
|
| 6 |
def add_op_namespace_prefix(op_name: str):
|
| 7 |
"""
|
| 8 |
Prefix op by namespace.
|
| 9 |
"""
|
| 10 |
+
return f"_cv_utils_19700101000000::{op_name}"
|
build/torch29-cxx11-cu128-x86_64-linux/sam3_kernels/_ops.py
CHANGED
|
@@ -1,9 +1,10 @@
|
|
| 1 |
import torch
|
| 2 |
-
|
| 3 |
-
ops = torch.ops.
|
|
|
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
-
return f"
|
|
|
|
| 1 |
import torch
|
| 2 |
+
|
| 3 |
+
ops = torch.ops._cv_utils_19700101000000
|
| 4 |
+
|
| 5 |
|
| 6 |
def add_op_namespace_prefix(op_name: str):
|
| 7 |
"""
|
| 8 |
Prefix op by namespace.
|
| 9 |
"""
|
| 10 |
+
return f"_cv_utils_19700101000000::{op_name}"
|
build/torch29-cxx11-cu130-x86_64-linux/sam3_kernels/_ops.py
CHANGED
|
@@ -1,9 +1,10 @@
|
|
| 1 |
import torch
|
| 2 |
-
|
| 3 |
-
ops = torch.ops.
|
|
|
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
-
return f"
|
|
|
|
| 1 |
import torch
|
| 2 |
+
|
| 3 |
+
ops = torch.ops._cv_utils_19700101000000
|
| 4 |
+
|
| 5 |
|
| 6 |
def add_op_namespace_prefix(op_name: str):
|
| 7 |
"""
|
| 8 |
Prefix op by namespace.
|
| 9 |
"""
|
| 10 |
+
return f"_cv_utils_19700101000000::{op_name}"
|