zdevito 8a445f7bd5 Serve multistream graph captures from correct pool (#114647) ... This fixes #114320 by placing the logic for determining whether to allocate to a pool inside a callback that is controlled by CUDAGraph.cpp or by the python bound api to allocate a stream directly to a pool. Pull Request resolved: https://github.com/pytorch/pytorch/pull/114647 Approved by: https://github.com/ngimel, https://github.com/eellison		2023-12-18 18:24:15 +00:00
..
_dynamo	[dynamo] Make {guards,eval_frame}.py pass follow_imports typechecking (#113721)	2023-11-17 18:24:21 +00:00
__init__.pyi.in	Serve multistream graph captures from correct pool (#114647)	2023-12-18 18:24:15 +00:00
_autograd.pyi	Enable import following in MYPYNOFOLLOW (now MYPYINDUCTOR) (#113830)	2023-11-17 18:24:21 +00:00
_cpu.pyi	Quantization oneDNN backend only support VNNI CPU (#103653)	2023-06-19 09:50:07 +00:00
_cudnn.pyi	[3/3] Update .pyi Python stub files and enable 'UFMT' linter (#95268)	2023-03-01 23:50:56 +00:00
_distributed_autograd.pyi	[3/3] Update .pyi Python stub files and enable 'UFMT' linter (#95268)	2023-03-01 23:50:56 +00:00
_distributed_c10d.pyi	[c10d] Create a python c10d API _set_pg_timeout to set timeout (#115453)	2023-12-12 20:52:43 +00:00
_distributed_rpc_testing.pyi	[BE]: Apply ruff PERF fixes to torch (#104917)	2023-07-11 20:45:21 +00:00
_distributed_rpc.pyi	Enable import following in MYPYNOFOLLOW (now MYPYINDUCTOR) (#113830)	2023-11-17 18:24:21 +00:00
_functions.pyi	[3/3] Update .pyi Python stub files and enable 'UFMT' linter (#95268)	2023-03-01 23:50:56 +00:00
_functorch.pyi	Reland fakify FunctionalTensor (#107569)	2023-08-22 15:46:25 +00:00
_itt.pyi	Fix ITT unit-tests if PyTorch is compiled with USE_ITT=OFF (#86199)	2022-10-04 21:57:05 +00:00
_lazy_ts_backend.pyi	[3/3] Update .pyi Python stub files and enable 'UFMT' linter (#95268)	2023-03-01 23:50:56 +00:00
_lazy.pyi	[3/3] Update .pyi Python stub files and enable 'UFMT' linter (#95268)	2023-03-01 23:50:56 +00:00
_monitor.pyi	[3/3] Update .pyi Python stub files and enable 'UFMT' linter (#95268)	2023-03-01 23:50:56 +00:00
_nn.pyi.in	Enable import following in MYPYNOFOLLOW (now MYPYINDUCTOR) (#113830)	2023-11-17 18:24:21 +00:00
_nvtx.pyi	[PyTorch] Support NVTX range_start and range_end (#70030)	2022-02-07 17:31:57 +00:00
_onnx.pyi	[ONNX] Add initial support for FP8 ONNX export (#107962)	2023-09-08 20:40:39 +00:00
_profiler.pyi	Enable import following in MYPYNOFOLLOW (now MYPYINDUCTOR) (#113830)	2023-11-17 18:24:21 +00:00
_VariableFunctions.pyi.in	Enable import following in MYPYNOFOLLOW (now MYPYINDUCTOR) (#113830)	2023-11-17 18:24:21 +00:00
_verbose.pyi	[RFC] enable oneMKL&oneDNN on-demands verbose functinality (#63212)	2022-07-27 23:29:35 +00:00
build.bzl	Make all .pyi.in files exportable from torch/_C/ folder (#74962)	2022-03-31 12:54:14 +00:00
return_types.pyi.in	Change flash attention outputs to be SymInt instead of int (#110533)	2023-10-05 01:00:07 +00:00