13 lines
401 B
Python
13 lines
401 B
Python
# 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).read(), filepath, mode='exec'))
|