Skip to content

Commit 901a4f5

Browse files
committed
test(cuda.core): isolate NUMA regression subprocess
Signed-off-by: Aryan <[email protected]>
1 parent 6efba07 commit 901a4f5

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

‎cuda_core/tests/test_device.py‎

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
# SPDX-License-Identifier: Apache-2.0
33

44
import contextlib
5+
import os
56
import subprocess
67
import sys
78
import textwrap
@@ -474,7 +475,7 @@ def test_device_properties_complete():
474475

475476
@pytest.mark.agent_authored(model="gpt-6")
476477
@pytest.mark.parametrize("driver_version", [12050, 12060, 12080])
477-
def test_host_numa_memory_pools_supported_on_pre_12_9_driver(driver_version):
478+
def test_host_numa_memory_pools_supported_on_pre_12_9_driver(driver_version, tmp_path):
478479
script = textwrap.dedent(
479480
"""
480481
import sys
@@ -509,8 +510,11 @@ def test_host_numa_memory_pools_supported_on_pre_12_9_driver(driver_version):
509510
assert properties.host_numa_memory_pools_supported is expected
510511
"""
511512
)
513+
env = {key: value for key, value in os.environ.items() if key != "PYTHONPATH"}
512514
result = subprocess.run( # noqa: S603 - executes a repository-owned test script
513515
[sys.executable, "-c", script, str(driver_version)],
516+
cwd=tmp_path,
517+
env=env,
514518
capture_output=True,
515519
check=False,
516520
text=True,

0 commit comments

Comments
 (0)