Skip to content

PR #43897: [ROCm] Don't use async deallocaton in MIOpen autotuner backend#120640

Merged
copybara-service[bot] merged 1 commit into
masterfrom
exported_pr_928672807
Jun 8, 2026
Merged

PR #43897: [ROCm] Don't use async deallocaton in MIOpen autotuner backend#120640
copybara-service[bot] merged 1 commit into
masterfrom
exported_pr_928672807

Conversation

@copybara-service

Copy link
Copy Markdown

PR #43897: [ROCm] Don't use async deallocaton in MIOpen autotuner backend

Imported from GitHub PR openxla/xla#43897

📝 Summary of Changes
Introduce file local OwningScratchAllocator implementation that does deallocation on destruction.

🎯 Justification
OwningScratchAllocator implementation moved to async deallocaton model which doesn't work for miopen backend.

🚀 Kind of Contribution
🐛 Bug Fix
Copybara import of the project:

--
945b5c2767fc40f51f9c045ba8766da3d728785c by Dragan Mladjenovic Dragan.Mladjenovic@amd.com:

[ROCm] Don't use async deallocaton in MIOpen autotuner backend

Merging this change closes #43897

FUTURE_COPYBARA_INTEGRATE_REVIEW=openxla/xla#43897 from ROCm:draganm/miopen_scratch_allocator 945b5c2767fc40f51f9c045ba8766da3d728785c

@copybara-service copybara-service Bot force-pushed the exported_pr_928672807 branch from d79bac4 to ee7a25e Compare June 8, 2026 20:18
…kend

Imported from GitHub PR openxla/xla#43897

📝 Summary of Changes
Introduce file local OwningScratchAllocator implementation that does deallocation on destruction.

🎯 Justification
OwningScratchAllocator implementation moved to async deallocaton model which doesn't work for miopen backend.

🚀 Kind of Contribution
🐛 Bug Fix
Copybara import of the project:

--
945b5c2767fc40f51f9c045ba8766da3d728785c by Dragan Mladjenovic <Dragan.Mladjenovic@amd.com>:

[ROCm] Don't use async deallocaton in MIOpen autotuner backend

Merging this change closes #43897

PiperOrigin-RevId: 928753564
@copybara-service copybara-service Bot force-pushed the exported_pr_928672807 branch from ee7a25e to bb81f08 Compare June 8, 2026 20:57
@copybara-service copybara-service Bot merged commit bb81f08 into master Jun 8, 2026
1 check passed
@copybara-service copybara-service Bot deleted the exported_pr_928672807 branch June 8, 2026 20:57
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.

Custom Op Output Shape is Unknown in TF 1.15

1 participant