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.
Files changed (56) hide show
  1. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/PKG-INFO +9 -2
  2. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/pyproject.toml +29 -2
  3. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/rust_core/Cargo.lock +335 -78
  4. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/rust_core/Cargo.toml +7 -7
  5. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/rust_core/src/backend_cpu.rs +30 -20
  6. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/rust_core/src/backend_gpu.rs +9 -9
  7. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/rust_core/src/lib.rs +12 -10
  8. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/rust_core/src/main.rs +1 -1
  9. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/rust_core/src/mmap_arrow.rs +3 -7
  10. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/backends/ast_backend.py +32 -39
  11. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/backends/cpu_backend.py +34 -16
  12. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/backends/cudf_backend.py +36 -3
  13. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/backends/cybert_backend.py +27 -9
  14. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/backends/ripgrep_backend.py +76 -38
  15. tensor_grep-0.3.4/src/tensor_grep/backends/torch_backend.py +104 -0
  16. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/cli/formatters/ripgrep_fmt.py +1 -1
  17. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/cli/main.py +210 -33
  18. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/cli/mcp_server.py +2 -1
  19. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/core/config.py +2 -0
  20. tensor_grep-0.3.4/src/tensor_grep/core/observability.py +16 -0
  21. tensor_grep-0.3.4/src/tensor_grep/core/pipeline.py +180 -0
  22. tensor_grep-0.3.4/src/tensor_grep/perf_guard.py +46 -0
  23. tensor_grep-0.2.1/src/tensor_grep/backends/torch_backend.py +0 -187
  24. tensor_grep-0.2.1/src/tensor_grep/core/pipeline.py +0 -109
  25. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/rust_core/src/backend_ast.rs +0 -0
  26. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/rust_core/src/cli.rs +0 -0
  27. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/rust_core/tests/test_output.rs +0 -0
  28. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/rust_core/tests/test_replace.rs +0 -0
  29. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/rust_core/tests/test_search.rs +0 -0
  30. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/__init__.py +0 -0
  31. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/backends/__init__.py +0 -0
  32. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/backends/ast_wrapper_backend.py +0 -0
  33. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/backends/base.py +0 -0
  34. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/backends/rust_backend.py +0 -0
  35. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/backends/stringzilla_backend.py +0 -0
  36. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/cli/__init__.py +0 -0
  37. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/cli/formatters/__init__.py +0 -0
  38. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/cli/formatters/base.py +0 -0
  39. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/cli/formatters/csv_fmt.py +0 -0
  40. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/cli/formatters/json_fmt.py +0 -0
  41. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/cli/formatters/table_fmt.py +0 -0
  42. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/cli/lsp_server.py +0 -0
  43. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/core/__init__.py +0 -0
  44. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/core/hardware/__init__.py +0 -0
  45. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/core/hardware/device_detect.py +0 -0
  46. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/core/hardware/memory_manager.py +0 -0
  47. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/core/profile_stats +0 -0
  48. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/core/query_analyzer.py +0 -0
  49. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/core/result.py +0 -0
  50. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/io/__init__.py +0 -0
  51. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/io/base.py +0 -0
  52. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/io/directory_scanner.py +0 -0
  53. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/io/reader_cudf.py +0 -0
  54. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/io/reader_dstorage.py +0 -0
  55. {tensor_grep-0.2.1 → tensor_grep-0.3.4}/src/tensor_grep/io/reader_fallback.py +0 -0
  56. {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.2.1
4
- Requires-Dist: typer[all]>=0.12
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.2.1"
13
+ version = "0.3.4"
14
14
  requires-python = ">=3.11"
15
15
  dependencies = [
16
- "typer[all]>=0.12",
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
+ ]