frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Graph break on tostring for numpy remapping (#131007)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/131007
Approved by: https://github.com/williamwen42
This commit is contained in:
Michael Lazos
2024-07-17 19:16:14 -07:00
committed by PyTorch MergeBot
parent 8bf0be7c78
commit 22388ffe03
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
torch/_dynamo/variables/tensor.py
View File

@ -1171,8 +1171,8 @@ class NumpyNdarrayVariable(TensorVariable):
if name in ["__len__", "size", "tolist"]:
# delegate back to TensorVariable
return super().call_method(tx, name, args, kwargs)
if name == "tobytes":
unimplemented("tobytes is not modelled in torch._numpy")
if name in ("tostring", "tobytes"):
unimplemented(f"{name} is not modelled in torch._numpy")
proxy = tx.output.create_proxy(
"call_function",
numpy_method_wrapper(name),