This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
AI流水线
Repositories
Datasets
Forum
实训
竞赛
大数据
AI开发
Register
Sign In
wangwei
/
MegEngine
Not watched
Unwatch
Watch all
Watch but not notify
1
Star
0
Fork
0
Code
Releases
31
Wiki
Activity
Issues
0
Pull Requests
0
Datasets
Model
Cloudbrain
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
4 years ago
parent
66c18f6054
commit
ebb7f5cac2
1 changed files
with
4 additions
and
0 deletions
Unified View
Diff Options
Show Stats
Download Patch File
Download Diff File
+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")
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():
def test_conv_transpose2d():
rng = np.random.RandomState(seed=2021)
rng = np.random.RandomState(seed=2021)
Write
Preview
Loading…
Cancel
Save