Edward Z. Yang f19e07b056 Memoize local_scalar_dense calls, refactor all memos (#125623) ... Signed-off-by: Edward Z. Yang <ezyang@meta.com> Pull Request resolved: https://github.com/pytorch/pytorch/pull/125623 Approved by: https://github.com/eellison		2024-05-11 21:12:35 +00:00
..
__init__.py
fake_impls.py	Memoize local_scalar_dense calls, refactor all memos (#125623)	2024-05-11 21:12:35 +00:00
fake_tensor.py	Memoize local_scalar_dense calls, refactor all memos (#125623)	2024-05-11 21:12:35 +00:00
fake_utils.py	Revert "[fake_impls] Fix seed/offset device for attention kernels (#120839)" (#121447)	2024-03-08 01:48:23 +00:00
functional_tensor.py	[RELAND] Switch default behavoir of export IR to be predispatch (#125860)	2024-05-10 17:36:53 +00:00
meta_utils.py	functionalize storage resizing, minimal ppFSDP traceable forward (#122434)	2024-05-10 18:09:10 +00:00
schema_check_mode.py	Replace follow_imports = silent with normal (#118414)	2024-01-27 02:44:11 +00:00