Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[SOT][PIR] support numpy and register_hook attr #66008

Merged
merged 12 commits into from
Jul 19, 2024
Prev Previous commit
Next Next commit
clean FallbackError
  • Loading branch information
gouzil committed Jul 13, 2024
commit 63de0057f2d57988f5ff101a13427cb03575d36a
Original file line number Diff line number Diff line change
Expand Up @@ -586,8 +586,6 @@ def getattr(self, name: str, default=None):
builtin_fn, self.graph, DanglingTracker()
).bind(self, name)
elif name in get_tensor_methods():
if name in ["numpy", "regisiter_hook"]:
raise FallbackError(f"no support {name}.")
from .callable import TensorFunctionVariable

fn_var = TensorFunctionVariable(
Expand Down