mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
9945fd7253
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/49980 From ``` ./python/libcst/libcst codemod remove_unused_imports.RemoveUnusedImportsWithGlean --no-format caffe2/ ``` Test Plan: Standard sandcastle tests Reviewed By: xush6528 Differential Revision: D25727359 fbshipit-source-id: c4f60005b10546423dc093d31d46deb418352286
38 lines
1019 B
Python
38 lines
1019 B
Python
|
|
|
|
|
|
|
|
|
|
import unittest
|
|
import hypothesis.strategies as st
|
|
from hypothesis import given
|
|
from caffe2.python import core, workspace
|
|
import caffe2.python.hypothesis_test_util as hu
|
|
import caffe2.python.mkl_test_util as mu
|
|
|
|
|
|
@unittest.skipIf(not workspace.C.has_mkldnn,
|
|
"Skipping as we do not have mkldnn.")
|
|
class MKLFillTest(hu.HypothesisTestCase):
|
|
@given(n=st.integers(1, 4), c=st.integers(1, 4),
|
|
h=st.integers(1, 4), w=st.integers(1, 4),
|
|
filler=st.sampled_from(
|
|
["XavierFill", "ConstantFill", "GaussianFill", "MSRAFill"]
|
|
),
|
|
seed=st.integers(5, 10),
|
|
**mu.gcs_cpu_mkl)
|
|
def test_mkl_fill(self, n, c, h, w, filler, seed, gc, dc):
|
|
op = core.CreateOperator(
|
|
filler,
|
|
[],
|
|
["Y"],
|
|
shape=[n, c, h, w],
|
|
)
|
|
for d in dc:
|
|
d.random_seed = seed
|
|
self.assertDeviceChecks(dc, op, [], [0])
|
|
|
|
if __name__ == "__main__":
|
|
import unittest
|
|
unittest.main()
|