Browse Source

test_get_padder添加jittor标签

tags/v1.0.0alpha
x54-729 2 years ago
parent
commit
dd83e0b2c7
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      tests/core/collators/padders/test_get_padder.py

+ 1
- 0
tests/core/collators/padders/test_get_padder.py View File

@@ -17,6 +17,7 @@ def test_get_element_shape_dtype():
@pytest.mark.parametrize('backend', ['raw', None, 'numpy', 'torch', 'jittor', 'paddle']) @pytest.mark.parametrize('backend', ['raw', None, 'numpy', 'torch', 'jittor', 'paddle'])
@pytest.mark.torch @pytest.mark.torch
@pytest.mark.paddle @pytest.mark.paddle
@pytest.mark.jittor
def test_get_padder_run(backend): def test_get_padder_run(backend):
if not _NEED_IMPORT_TORCH and backend == 'torch': if not _NEED_IMPORT_TORCH and backend == 'torch':
pytest.skip("No torch") pytest.skip("No torch")


Loading…
Cancel
Save