7 Commits

Author	SHA1	Message	Date
psychedelicious	c1cf01a038	tests: use dangerously_run_function_in_subprocess to fix configure_torch_cuda_allocator tests	2025-03-06 07:49:35 +11:00
psychedelicious	d193e4f02a	feat(app): log warning instead of raising if PYTORCH_CUDA_ALLOC_CONF is already set	2025-03-06 07:49:35 +11:00
psychedelicious	ec493e30ee	feat(app): make logger a required arg in configure_torch_cuda_allocator	2025-03-06 07:49:35 +11:00
Ryan Dick	a36a627f83	Switch from use_cuda_malloc flag to a general pytorch_cuda_alloc_conf config field that allows full customization of the CUDA allocator.	2025-02-28 21:39:09 +00:00
Ryan Dick	b31c71f302	Simplify is_torch_cuda_malloc_enabled() implementation and add unit tests.	2025-02-28 21:39:09 +00:00
Ryan Dick	5302d4890f	Add use_cuda_malloc config option.	2025-02-28 21:39:09 +00:00
Ryan Dick	766b752572	Add utils for configuring the torch CUDA allocator.	2025-02-28 21:39:09 +00:00