tensor-grep 0.2.1__tar.gz → 0.3.4__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/PKG-INFO +9 -2
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/pyproject.toml +29 -2
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/rust_core/Cargo.lock +335 -78
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/rust_core/Cargo.toml +7 -7
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/rust_core/src/backend_cpu.rs +30 -20
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/rust_core/src/backend_gpu.rs +9 -9
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/rust_core/src/lib.rs +12 -10
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/rust_core/src/main.rs +1 -1
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/rust_core/src/mmap_arrow.rs +3 -7
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/backends/ast_backend.py +32 -39
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/backends/cpu_backend.py +34 -16
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/backends/cudf_backend.py +36 -3
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/backends/cybert_backend.py +27 -9
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/backends/ripgrep_backend.py +76 -38
- tensor_grep-0.3.4/src/tensor_grep/backends/torch_backend.py +104 -0
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/cli/formatters/ripgrep_fmt.py +1 -1
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/cli/main.py +210 -33
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/cli/mcp_server.py +2 -1
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/core/config.py +2 -0
- tensor_grep-0.3.4/src/tensor_grep/core/observability.py +16 -0
- tensor_grep-0.3.4/src/tensor_grep/core/pipeline.py +180 -0
- tensor_grep-0.3.4/src/tensor_grep/perf_guard.py +46 -0
- tensor_grep-0.2.1/src/tensor_grep/backends/torch_backend.py +0 -187
- tensor_grep-0.2.1/src/tensor_grep/core/pipeline.py +0 -109
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/rust_core/src/backend_ast.rs +0 -0
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/rust_core/src/cli.rs +0 -0
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/rust_core/tests/test_output.rs +0 -0
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/rust_core/tests/test_replace.rs +0 -0
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/rust_core/tests/test_search.rs +0 -0
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/__init__.py +0 -0
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/backends/__init__.py +0 -0
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/backends/ast_wrapper_backend.py +0 -0
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/backends/base.py +0 -0
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/backends/rust_backend.py +0 -0
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/backends/stringzilla_backend.py +0 -0
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/cli/__init__.py +0 -0
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/cli/formatters/__init__.py +0 -0
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/cli/formatters/base.py +0 -0
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/cli/formatters/csv_fmt.py +0 -0
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/cli/formatters/json_fmt.py +0 -0
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/cli/formatters/table_fmt.py +0 -0
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/cli/lsp_server.py +0 -0
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/core/__init__.py +0 -0
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/core/hardware/__init__.py +0 -0
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/core/hardware/device_detect.py +0 -0
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/core/hardware/memory_manager.py +0 -0
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/core/profile_stats +0 -0
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/core/query_analyzer.py +0 -0
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/core/result.py +0 -0
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/io/__init__.py +0 -0
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/io/base.py +0 -0
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/io/directory_scanner.py +0 -0
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/io/reader_cudf.py +0 -0
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/io/reader_dstorage.py +0 -0
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/io/reader_fallback.py +0 -0
- {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/io/reader_kvikio.py +0 -0
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
Metadata-Version: 2.4
|
|
2
2
|
Name: tensor-grep
|
|
3
|
-
Version: 0.
|
|
4
|
-
Requires-Dist: typer
|
|
3
|
+
Version: 0.3.4
|
|
4
|
+
Requires-Dist: typer>=0.12
|
|
5
5
|
Requires-Dist: rich>=13.0
|
|
6
6
|
Requires-Dist: mcp>=1.2.0
|
|
7
7
|
Requires-Dist: pyarrow>=14.0.0
|
|
@@ -13,6 +13,12 @@ Requires-Dist: tree-sitter>=0.22 ; extra == 'ast'
|
|
|
13
13
|
Requires-Dist: tree-sitter-python ; extra == 'ast'
|
|
14
14
|
Requires-Dist: tree-sitter-javascript ; extra == 'ast'
|
|
15
15
|
Requires-Dist: pygls>=1.3.0 ; extra == 'ast'
|
|
16
|
+
Requires-Dist: pytest>=8.0 ; extra == 'bench'
|
|
17
|
+
Requires-Dist: tree-sitter>=0.22 ; extra == 'bench'
|
|
18
|
+
Requires-Dist: tree-sitter-python ; extra == 'bench'
|
|
19
|
+
Requires-Dist: tree-sitter-javascript ; extra == 'bench'
|
|
20
|
+
Requires-Dist: torch>=2.0 ; extra == 'bench'
|
|
21
|
+
Requires-Dist: nvtx>=0.2 ; extra == 'bench'
|
|
16
22
|
Requires-Dist: pytest>=8.0 ; extra == 'dev'
|
|
17
23
|
Requires-Dist: pytest-cov>=5.0 ; extra == 'dev'
|
|
18
24
|
Requires-Dist: pytest-mock>=3.14 ; extra == 'dev'
|
|
@@ -35,6 +41,7 @@ Requires-Dist: torch-geometric>=2.5.0 ; extra == 'gpu-win'
|
|
|
35
41
|
Requires-Dist: transformers>=4.40 ; extra == 'nlp'
|
|
36
42
|
Requires-Dist: tritonclient[all] ; extra == 'nlp'
|
|
37
43
|
Provides-Extra: ast
|
|
44
|
+
Provides-Extra: bench
|
|
38
45
|
Provides-Extra: dev
|
|
39
46
|
Provides-Extra: gpu
|
|
40
47
|
Provides-Extra: gpu-win
|
|
@@ -10,10 +10,10 @@ manifest-path = "rust_core/Cargo.toml"
|
|
|
10
10
|
|
|
11
11
|
[project]
|
|
12
12
|
name = "tensor-grep"
|
|
13
|
-
version = "0.
|
|
13
|
+
version = "0.3.4"
|
|
14
14
|
requires-python = ">=3.11"
|
|
15
15
|
dependencies = [
|
|
16
|
-
"typer
|
|
16
|
+
"typer>=0.12",
|
|
17
17
|
"rich>=13.0",
|
|
18
18
|
"mcp>=1.2.0",
|
|
19
19
|
"pyarrow>=14.0.0",
|
|
@@ -38,6 +38,14 @@ dev = [
|
|
|
38
38
|
"psutil",
|
|
39
39
|
"types-PyYAML",
|
|
40
40
|
]
|
|
41
|
+
bench = [
|
|
42
|
+
"pytest>=8.0",
|
|
43
|
+
"tree-sitter>=0.22",
|
|
44
|
+
"tree-sitter-python",
|
|
45
|
+
"tree-sitter-javascript",
|
|
46
|
+
"torch>=2.0",
|
|
47
|
+
"nvtx>=0.2",
|
|
48
|
+
]
|
|
41
49
|
|
|
42
50
|
[project.scripts]
|
|
43
51
|
tg = "tensor_grep.cli.main:main_entry"
|
|
@@ -76,6 +84,14 @@ runner = "python -m pytest tests/unit/ -x --no-header -q"
|
|
|
76
84
|
[tool.ruff]
|
|
77
85
|
target-version = "py311"
|
|
78
86
|
line-length = 100
|
|
87
|
+
exclude = [
|
|
88
|
+
"bench_data",
|
|
89
|
+
"bench_ast_data",
|
|
90
|
+
"gpu_bench_data",
|
|
91
|
+
"benchmarks/bench_data",
|
|
92
|
+
"benchmarks/bench_ast_data",
|
|
93
|
+
"benchmarks/gpu_bench_data",
|
|
94
|
+
]
|
|
79
95
|
|
|
80
96
|
[tool.ruff.lint]
|
|
81
97
|
select = [
|
|
@@ -103,3 +119,14 @@ strict = true
|
|
|
103
119
|
ignore_missing_imports = true
|
|
104
120
|
warn_unused_configs = true
|
|
105
121
|
disallow_untyped_defs = true
|
|
122
|
+
|
|
123
|
+
[tool.semantic_release]
|
|
124
|
+
version_toml = ["pyproject.toml:project.version"]
|
|
125
|
+
branch = "main"
|
|
126
|
+
build_command = "curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y --profile minimal && . $HOME/.cargo/env && rustup default stable && pip install uv && uv build"
|
|
127
|
+
upload_to_pypi = false
|
|
128
|
+
upload_to_release = true
|
|
129
|
+
commit_message = "chore(release): v{version} [skip ci]"
|
|
130
|
+
version_variables = [
|
|
131
|
+
"src/tensor_grep/cli/main.py:pkg_version",
|
|
132
|
+
]
|