Skip to content

Commit 48211a7

Browse files
[release/2.11] Fix Windows access violation in MIOpen CTC loss dispatch (#3161)
Cherry pick of pytorch#178284 Fixes ROCm/TheRock#3987 Co-authored-by: Milica Stankovic <milica.stankovic@amd.com>
1 parent 3aaa914 commit 48211a7

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

aten/src/ATen/native/miopen/LossCTC_miopen.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
#include <ATen/ops/empty_like.h>
1313
#include <ATen/ops/miopen_ctc_loss.h>
1414
#include <ATen/ops/miopen_ctc_loss_native.h>
15+
#include <ATen/ops/_use_miopen_ctc_loss_native.h>
1516
#endif
1617

1718
// TODO: Remove the condition on AT_ROCM_ENABLED entirely,

0 commit comments

Comments
 (0)