mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-06-26 03:00:54 +00:00
Disable test_aten_conv_bf16 to unblock amd ci (#20499)
### Description <!-- Describe your changes. --> ### Motivation and Context <!-- - Why is this change required? What problem does it solve? - If it fixes an open issue, please link to the issue here. -->
This commit is contained in:
parent
b789677e65
commit
f31486c8b7
1 changed files with 2 additions and 0 deletions
|
|
@ -6610,6 +6610,8 @@ def test_overridden_softmax_export(softmax_compute_type):
|
|||
assert to_value == pytorch_type_to_onnx_dtype(softmax_compute_type), "Cast to attribute is not as expected"
|
||||
|
||||
|
||||
# TODO: fix the issue in rocm training, then enable the test.
|
||||
@pytest.mark.skip(reason="This test is disabled due to its breaking rocm training cis.")
|
||||
def test_aten_conv_bf16():
|
||||
class NeuralNetConv(torch.nn.Module):
|
||||
def __init__(self):
|
||||
|
|
|
|||
Loading…
Reference in a new issue