mirror of
git://git.yoctoproject.org/meta-intel.git
synced 2025-07-19 21:09:03 +02:00
0c092dde58
Add automated tests that: - test that mkl_dnn shared libraries and headers can be used to compile mkl_dnn sample application and execute it - test that mkl_dnn various api (eg. convolution, deconv, rnn, etc) are executable - add extra packages checking for test compiling mkldnn application Signed-off-by: Yeoh Ee Peng <ee.peng.yeoh@intel.com> Signed-off-by: Anuj Mittal <anuj.mittal@intel.com>
68 lines
3.2 KiB
Python
68 lines
3.2 KiB
Python
from oeqa.runtime.case import OERuntimeTestCase
|
|
from oeqa.runtime.decorator.package import OEHasPackage
|
|
from oeqa.core.decorator.depends import OETestDepends
|
|
from oeqa.runtime.miutils.targets.oeqatarget import OEQATarget
|
|
from oeqa.runtime.miutils.tests.mkl_dnn_test import MkldnnTest
|
|
|
|
class MklDnn(OERuntimeTestCase):
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
cls.mkldnntest = MkldnnTest(OEQATarget(cls.tc.target))
|
|
|
|
@classmethod
|
|
def tearDownClass(cls):
|
|
cls.mkldnntest.tear_down()
|
|
|
|
@OEHasPackage(['libdnnl', 'libdnnl1'])
|
|
@OEHasPackage(['libdnnl-src'])
|
|
@OEHasPackage(['libdnnl-dev'])
|
|
@OEHasPackage(['gcc'])
|
|
@OEHasPackage(['gcc-symlinks'])
|
|
@OEHasPackage(['libstdc++-dev'])
|
|
@OEHasPackage(['binutils'])
|
|
def test_mkldnn_can_compile_and_execute(self):
|
|
(status, output) = self.mkldnntest.test_mkldnn_can_compile_and_execute()
|
|
self.assertEqual(status, 0, msg='status and output: %s and %s' % (status, output))
|
|
|
|
@OEHasPackage(['mkl-dnn', 'libdnnl1'])
|
|
@OEHasPackage(['mkl-dnn-test', 'libdnnl-test'])
|
|
def test_mkldnn_benchdnn_package_available(self):
|
|
(status, output) = self.mkldnntest.test_mkldnn_benchdnn_package_available()
|
|
self.assertEqual(status, 0, msg='status and output: %s and %s' % (status, output))
|
|
|
|
@OETestDepends(['mkl_dnn.MklDnn.test_mkldnn_benchdnn_package_available'])
|
|
def test_mkldnn_conv_api(self):
|
|
(status, output) = self.mkldnntest.test_mkldnn_conv_api()
|
|
self.assertEqual(status, 0, msg='status and output: %s and %s' % (status, output))
|
|
|
|
@OETestDepends(['mkl_dnn.MklDnn.test_mkldnn_benchdnn_package_available'])
|
|
def test_mkldnn_bnorm_api(self):
|
|
(status, output) = self.mkldnntest.test_mkldnn_bnorm_api()
|
|
self.assertEqual(status, 0, msg='status and output: %s and %s' % (status, output))
|
|
|
|
@OETestDepends(['mkl_dnn.MklDnn.test_mkldnn_benchdnn_package_available'])
|
|
def test_mkldnn_deconv_api(self):
|
|
(status, output) = self.mkldnntest.test_mkldnn_deconv_api()
|
|
self.assertEqual(status, 0, msg='status and output: %s and %s' % (status, output))
|
|
|
|
@OETestDepends(['mkl_dnn.MklDnn.test_mkldnn_benchdnn_package_available'])
|
|
def test_mkldnn_ip_api(self):
|
|
(status, output) = self.mkldnntest.test_mkldnn_ip_api()
|
|
self.assertEqual(status, 0, msg='status and output: %s and %s' % (status, output))
|
|
|
|
@OETestDepends(['mkl_dnn.MklDnn.test_mkldnn_benchdnn_package_available'])
|
|
def test_mkldnn_reorder_api(self):
|
|
(status, output) = self.mkldnntest.test_mkldnn_reorder_api()
|
|
self.assertEqual(status, 0, msg='status and output: %s and %s' % (status, output))
|
|
|
|
@OETestDepends(['mkl_dnn.MklDnn.test_mkldnn_benchdnn_package_available'])
|
|
def test_mkldnn_rnn_api(self):
|
|
(status, output) = self.mkldnntest.test_mkldnn_rnn_api()
|
|
self.assertEqual(status, 0, msg='status and output: %s and %s' % (status, output))
|
|
|
|
@OETestDepends(['mkl_dnn.MklDnn.test_mkldnn_benchdnn_package_available'])
|
|
def test_mkldnn_shuffle_api(self):
|
|
(status, output) = self.mkldnntest.test_mkldnn_shuffle_api()
|
|
self.assertEqual(status, 0, msg='status and output: %s and %s' % (status, output))
|