blob: 31478cb6e05396c13f2b10f89a2ff5aae832e5b3 [file] [log] [blame]
# Owner(s): ["module: cuda"]
# run time cuda tests, but with the allocator using expandable segments
import os
import torch
if torch.cuda.is_available():
torch.cuda.memory._set_allocator_settings('expandable_segments:True')
current_dir = os.path.dirname(os.path.abspath(__file__))
filepath = os.path.join(current_dir, 'test_cuda.py')
exec(compile(open(filepath, 'r').read(), filepath, mode='exec'))