Skip to content

TestNN.test_cpu_vs_cudnn_no_dropout does not test MIOpen RNN on ROCm #34615

@ashishfarmer

Description

@ashishfarmer

Running PYTORCH_TEST_WITH_ROCM=1 python test_nn.py TestNN.test_rnn_cpu_vs_cudnn_no_dropout does not run MIOpen path.
This line in test_nn.py sets data type of tensors for the test to double, which is not supported by MIOpen which is why during execution we fall back to HIP kernels rather than using MIOpen. The change came from PR #27628 where the default datatype of the test was changed from float to double. Can we get a little more clarity on this change or if there could be a way to run MIOpen RNN in the CI tests.

cc: @mruberry @ezyang @iotamudelta

Metadata

Metadata

Assignees

No one assigned

    Labels

    module: testsIssues related to tests (not the torch.testing module)triagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate module

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions