2 commits

Author	SHA1	Message	Date
Philip Sinitsin	76c10cd635	jit: don't memplan buffers reachable from live tensors (#16588) ... The memory planner was suballocating BUFFERs created during JIT capture that are still referenced by external lazy tensor graphs, like the .grad tensors assigned by backward(). The replay then only writes the arena slices, so realizing such a tensor after the call reads freshly allocated memory and silently returns zeros. Hold every BUFFER reachable from a live Tensor instead of only the parameters of the return value; true internals are still planned. Fixes #16571.	2026-06-12 17:51:54 +03:00
Christopher Milan	9b0f75622c	many jit tests belong in unit (#16508)	2026-06-04 21:36:53 -04:00