Skip to content

[release/2.15][ROCm] Fix test_region_device_time and test_load_standalone on ROCm wheels (#201771) - #3869

Merged
pragupta merged 1 commit into
release/2.15from
dnikolaev/cherry-pick-201771-2.15
Oct 27, 2026
Merged

pragupta merged 1 commit into
release/2.15from
dnikolaev/cherry-pick-201771-2.15

Conversation

@dnikolaev-amd

Copy link
Copy Markdown

ROCm wheels setenv() ROCPROFILER_REGISTER_LIBRARY during GPU init. The child inherits it: test_region_device_time records no kernels, and the test_load_standalone binary aborts before main. Pass an environment rebuilt from os.environ, same as pytorch#197421.

Test Plan:

python test/profiler/test_profiler.py TestProfiler.test_region_device_time
python test/test_utils.py TestStandaloneCPPJIT.test_load_standalone

Pull Request resolved: pytorch#201771
Approved by: https://github.com/jeffdaily

(cherry picked from commit 95f55ae)

…heels (pytorch#201771)

ROCm wheels setenv() ROCPROFILER_REGISTER_LIBRARY during GPU init. The child inherits it: test_region_device_time records no kernels, and the test_load_standalone binary aborts before main. Pass an environment rebuilt from os.environ, same as pytorch#197421.

Test Plan:

```
python test/profiler/test_profiler.py TestProfiler.test_region_device_time
python test/test_utils.py TestStandaloneCPPJIT.test_load_standalone
```

Pull Request resolved: pytorch#201771
Approved by: https://github.com/jeffdaily

(cherry picked from commit 95f55ae)
@dnikolaev-amd dnikolaev-amd changed the title [ROCm] Fix test_region_device_time and test_load_standalone on ROCm wheels (#201771) [release/2.15][ROCm] Fix test_region_device_time and test_load_standalone on ROCm wheels (#201771) Oct 26, 2026
@pragupta
pragupta merged commit 6eaece6 into release/2.15 Oct 27, 2026
7 checks passed
@pragupta
pragupta deleted the dnikolaev/cherry-pick-201771-2.15 branch October 27, 2026 18:38
jithunnair-amd pushed a commit that referenced this pull request Nov 4, 2026
…lone on ROCm wheels (pytorch#201771) (#3869)

ROCm wheels setenv() ROCPROFILER_REGISTER_LIBRARY during GPU init. The
child inherits it: test_region_device_time records no kernels, and the
test_load_standalone binary aborts before main. Pass an environment
rebuilt from os.environ, same as pytorch#197421.

Test Plan:

```
python test/profiler/test_profiler.py TestProfiler.test_region_device_time
python test/test_utils.py TestStandaloneCPPJIT.test_load_standalone
```

Pull Request resolved: pytorch#201771
Approved by: https://github.com/jeffdaily

(cherry picked from commit 95f55ae)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants