saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add hex method on SymFloat (#139451)

Browse source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/139451
Approved by: https://github.com/ezyang
This commit is contained in:
Bob Ren 2024-11-01 14:06:50 -07:00 committed by PyTorch MergeBot
parent 8d1eaa3da6
commit fdd298dcb7
2 changed files with 7 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
torch/__init__.py
View file

@ -681,6 +681,10 @@ class SymFloat:
"""Returns the complex conjugate of the float."""
return self
def hex(self) -> str:
"""Returns the hexadecimal representation of the float."""
return self.node.guard_float("", 0).hex()
class SymBool:
"""

3
torch/_dynamo/variables/builder.py
View file

@ -2384,6 +2384,9 @@ def handle_traced_output(example_value, tx, proxy, options, subclass_type, targe
):
set_example_value(proxy.node, example_value)
return ConstantVariable.create(example_value, **options)
elif isinstance(example_value, str) and (proxy.node.target in ["hex"]):
set_example_value(proxy.node, example_value)
return ConstantVariable.create(example_value, **options)
else:
unimplemented(
"torch.* op returned non-Tensor "