mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
578388bed8
This reverts commit fdc83610f2.
Reverted https://github.com/pytorch/pytorch/pull/128068 on behalf of https://github.com/janeyx99 due to Reverting for breaking ROCm tests on trunk, I think the tests need to be qualified with @onlyCUDA ([comment](https://github.com/pytorch/pytorch/pull/128068#issuecomment-2223672381))
16 lines
491 B
Python
16 lines
491 B
Python
# Owner(s): ["module: cuda"]
|
|
# run time cuda tests, but with the allocator using expandable segments
|
|
|
|
import os
|
|
|
|
import torch
|
|
|
|
from torch.testing._internal.common_cuda import IS_JETSON
|
|
|
|
if torch.cuda.is_available() and not IS_JETSON:
|
|
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"))
|