Commit c643cb5e authored by fenglei's avatar fenglei

using NGRAPH_TEST so test can be disable for specific backend

parent 8fe23cae
...@@ -25,6 +25,7 @@ ...@@ -25,6 +25,7 @@
#include "ngraph/serializer.hpp" #include "ngraph/serializer.hpp"
#include "util/all_close_f.hpp" #include "util/all_close_f.hpp"
#include "util/random.hpp" #include "util/random.hpp"
#include "util/test_control.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ngraph;
...@@ -93,29 +94,29 @@ static void test_allreduce_common(reduction::Type reduce_type) ...@@ -93,29 +94,29 @@ static void test_allreduce_common(reduction::Type reduce_type)
} }
} }
TEST(distributed_${BACKEND_NAME}, allreduce_sum) NGRAPH_TEST(distributed_${BACKEND_NAME}, allreduce_sum)
{ {
test_allreduce_common(reduction::Type::SUM); test_allreduce_common(reduction::Type::SUM);
} }
TEST(distributed_${BACKEND_NAME}, allreduce_min) NGRAPH_TEST(distributed_${BACKEND_NAME}, allreduce_min)
{ {
test_allreduce_common(reduction::Type::MIN); test_allreduce_common(reduction::Type::MIN);
} }
TEST(distributed_${BACKEND_NAME}, allreduce_max) NGRAPH_TEST(distributed_${BACKEND_NAME}, allreduce_max)
{ {
test_allreduce_common(reduction::Type::MAX); test_allreduce_common(reduction::Type::MAX);
} }
#if !defined(NGRAPH_DISTRIBUTED_MLSL_ENABLE) #if !defined(NGRAPH_DISTRIBUTED_MLSL_ENABLE)
TEST(distributed_${BACKEND_NAME}, allreduce_prod) NGRAPH_TEST(distributed_${BACKEND_NAME}, allreduce_prod)
{ {
test_allreduce_common(reduction::Type::PROD); test_allreduce_common(reduction::Type::PROD);
} }
#endif #endif
TEST(distributed_${BACKEND_NAME}, broadcastdistributed) NGRAPH_TEST(distributed_${BACKEND_NAME}, broadcastdistributed)
{ {
auto shape = Shape{2, 2}; auto shape = Shape{2, 2};
auto A = make_shared<op::Parameter>(element::f32, shape); auto A = make_shared<op::Parameter>(element::f32, shape);
...@@ -145,7 +146,7 @@ TEST(distributed_${BACKEND_NAME}, broadcastdistributed) ...@@ -145,7 +146,7 @@ TEST(distributed_${BACKEND_NAME}, broadcastdistributed)
//MLSL does not support send recv //MLSL does not support send recv
#if !defined(NGRAPH_DISTRIBUTED_MLSL_ENABLE) #if !defined(NGRAPH_DISTRIBUTED_MLSL_ENABLE)
TEST(distributed_${BACKEND_NAME}, send_recv) NGRAPH_TEST(distributed_${BACKEND_NAME}, send_recv)
{ {
auto shape = Shape{2, 2}; auto shape = Shape{2, 2};
auto A = make_shared<op::Parameter>(element::f32, shape); auto A = make_shared<op::Parameter>(element::f32, shape);
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment