Browse Source

fix(functional): hide format interface and skip test_conv_bias

GitOrigin-RevId: 7de75bf084
tags/v1.0.0-rc1
Megvii Engine Team 4 years ago
parent
commit
b970c6daf1
2 changed files with 2 additions and 2 deletions
  1. +1
    -2
      imperative/python/megengine/functional/quantized.py
  2. +1
    -0
      imperative/python/test/unit/functional/test_functional.py

+ 1
- 2
imperative/python/megengine/functional/quantized.py View File

@@ -24,7 +24,6 @@ def conv_bias_activation(
padding: Union[int, Tuple[int, int]] = 0,
dilation: Union[int, Tuple[int, int]] = 1,
groups: int = 1,
format="NCHW",
nonlinear_mode="IDENTITY",
conv_mode="CROSS_CORRELATION",
compute_mode="DEFAULT",
@@ -69,7 +68,7 @@ def conv_bias_activation(
dilate_h=dh,
dilate_w=dw,
dtype=dtype,
format=format,
format="NCHW",
strategy=get_conv_execution_strategy(),
nonlineMode=nonlinear_mode,
mode=conv_mode,


+ 1
- 0
imperative/python/test/unit/functional/test_functional.py View File

@@ -417,6 +417,7 @@ def test_batched_nms():
np.testing.assert_equal(results.numpy(), np.array([1, 4, 5], dtype=np.int32))


@pytest.mark.skip(reason="cuda does not support nchw int8")
def test_conv_bias():
inp_scale = 1.5
w_scale = 2.5


Loading…
Cancel
Save