diff --git a/tests/test_benchmark.py b/tests/test_benchmark.py index 6e2dbf8f4c73950541bea240bea0ca2d81fb7be4..48207dad60960914081a5324e8873dceb76f4cc4 100755 --- a/tests/test_benchmark.py +++ b/tests/test_benchmark.py @@ -48,7 +48,8 @@ def test_generate_gpu(): a[0] @= b[0] + c[0] kernel_vadd = ps.create_kernel(**vadd) - temp_dir = Path('/tmp/pystencils_benchmark_testfolder') - generate_benchmark_gpu([kernel_vadd], temp_dir, compiler=compiler) - # subprocess.run(['make', '-C', f'{temp_dir}'], check=True) - # subprocess.run([f'{temp_dir}/benchmark-{compiler.name}', '10'], check=True) + with tempfile.TemporaryDirectory(dir=Path.cwd()) as temp_dir: + temp_dir = Path(temp_dir) + generate_benchmark_gpu(kernel_vadd, temp_dir, compiler=compiler) + # subprocess.run(['make', '-C', f'{temp_dir}'], check=True) + # subprocess.run([f'{temp_dir}/benchmark-{compiler.name}', '10'], check=True)