5 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	d037d8f9aa	tests: update tests for configure_torch_cuda_allocator	2025-03-06 07:49:35 +11:00
Ryan Dick	49191709a0	Mark test_configure_torch_cuda_allocator_raises_if_torch_is_already_imported() to only run if CUDA is available.	2025-02-28 21:39:09 +00: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