This is an automated email from the ASF dual-hosted git repository.
tqchen pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new 2fd72ab29f [TE] [FFI] Fix broken axis/reduce_axis properties in
BaseComputeOp and ScanOp after FFI refactoring (#18375)
2fd72ab29f is described below
commit 2fd72ab29f6247965e8805e34265872d4ba77bb5
Author: BenkangPeng <[email protected]>
AuthorDate: Fri Oct 17 03:10:27 2025 +0800
[TE] [FFI] Fix broken axis/reduce_axis properties in BaseComputeOp and
ScanOp after FFI refactoring (#18375)
[TE][FFI] Remove unused properties from BaseComputeOp and ScanOp classes in
tensor.py
---
python/tvm/te/tensor.py | 15 ---------------
1 file changed, 15 deletions(-)
diff --git a/python/tvm/te/tensor.py b/python/tvm/te/tensor.py
index 11084da0cc..4ef1b67969 100644
--- a/python/tvm/te/tensor.py
+++ b/python/tvm/te/tensor.py
@@ -131,16 +131,6 @@ class PlaceholderOp(Operation):
class BaseComputeOp(Operation):
"""Compute operation."""
- @property
- def axis(self):
- """Represent the IterVar axis, defined when it is a ComputeOp"""
- return self.__getattr__("axis")
-
- @property
- def reduce_axis(self):
- """Represent axis of reductions, only defined when it is a ComputeOp"""
- return self.__getattr__("reduce_axis")
-
@tvm_ffi.register_object("te.ComputeOp")
class ComputeOp(BaseComputeOp):
@@ -151,11 +141,6 @@ class ComputeOp(BaseComputeOp):
class ScanOp(Operation):
"""Scan operation."""
- @property
- def scan_axis(self):
- """Represent the scan axis, only defined when it is a ScanOp"""
- return self.__getattr__("scan_axis")
-
@tvm_ffi.register_object("te.ExternOp")
class ExternOp(Operation):
