feat(app): revised `configure_torch_cuda_allocator()` & testing strategy#7733

Merged

psychedelicious merged 7 commits intomainfrom psyche/feat/app/cuda-alloc-util

Mar 5, 2025

+241-14

Commits on Mar 5, 2025

feat(app): make logger a required arg in configure_torch_cuda_allocator
psychedelicious
committed
feat(app): log warning instead of raising if PYTORCH_CUDA_ALLOC_CONF is already set
psychedelicious
committed
tests: add testing utils to set/unset env var
psychedelicious
committed
tests: update tests for configure_torch_cuda_allocator
psychedelicious
committed
tests: add util to run a function in separate process
psychedelicious
committed
tests: use dangerously_run_function_in_subprocess to fix configure_torch_cuda_allocator tests
psychedelicious
committed
tidy: remove unused utils
psychedelicious
committed