Browse Source

fix(imperative/python/test): add device judgment for the test case of conv-transpose2d-quantized

GitOrigin-RevId: 456cb78439
tags/v1.6.0-rc1
Megvii Engine Team 3 years ago
parent
commit
ebb7f5cac2
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      imperative/python/test/unit/quantization/test_op.py

+ 4
- 0
imperative/python/test/unit/quantization/test_op.py View File

@@ -171,6 +171,10 @@ def test_conv_bias():
run(10, 36, 8, 46, 26, 2, 2, 2, 1, 1, 2, True, "relu")


@pytest.mark.skipif(
get_device_count("gpu") > 0,
reason="does not support int8 when gpu compute capability less than 6.1",
)
def test_conv_transpose2d():
rng = np.random.RandomState(seed=2021)



Loading…
Cancel
Save