Disable multiprocess test for importer.

kendryte · Apr 21, 2022 · ab7cd2c · ab7cd2c
1 parent e47fee3
commit ab7cd2c
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 12 deletions.
diff --git a/.github/workflows/compiler-test.yml b/.github/workflows/compiler-test.yml
@@ -103,7 +103,7 @@ jobs:
       run: |
 
         pytest -n 50 --dist=load tests/other --doctest-modules --junitxml=test_results/other.xml
-        pytest -n 50 --dist=load tests/importer --doctest-modules --junitxml=test_results/importer.xml
+        pytest --dist=load tests/importer --doctest-modules --junitxml=test_results/importer.xml
         pytest -n 50 --dist=load tests/schedule --doctest-modules --junitxml=test_results/schedule.xml
         pytest -n 50 --dist=load tests/graph_partition --doctest-modules --junitxml=test_results/graph_partition.xml
         pytest -n 50 --dist=load tests/transform --doctest-modules --junitxml=test_results/transform.xml

diff --git a/src/importer/tflite/ops/custom.cpp b/src/importer/tflite/ops/custom.cpp
@@ -27,17 +27,17 @@ DEFINE_TFLITE_LOWER(CUSTOM)
     std::cout << "custom_code = " << custom_code << std::endl;
     if (custom_code == "FlexRandomUniform")
     {
-        auto custom_options = op.custom_options();
-        auto r = flexbuffers::GetRoot(custom_options->data(), custom_options->size());
-        std::cout << "Reference type: " << r.GetType() << std::endl;
-        auto v = flexbuffers::GetRoot(custom_options->data(), custom_options->size()).AsVector();
-        std::cout << "v.size = " << v.size() << std::endl;
-        for (size_t i = 0; i < v.size(); i++)
-        {
-            std::cout << "i = " << i << ": Reference type: " << v[i].GetType() << std::endl;
-            if (v[i].IsString())
-                std::cout << v[i].AsString().str() << std::endl;
-        }
+        // auto custom_options = op.custom_options();
+        // auto r = flexbuffers::GetRoot(custom_options->data(), custom_options->size());
+        // std::cout << "Reference type: " << r.GetType() << std::endl;
+        // auto v = flexbuffers::GetRoot(custom_options->data(), custom_options->size()).AsVector();
+        // std::cout << "v.size = " << v.size() << std::endl;
+        // for (size_t i = 0; i < v.size(); i++)
+        // {
+        //     std::cout << "i = " << i << ": Reference type: " << v[i].GetType() << std::endl;
+        //     if (v[i].IsString())
+        //         std::cout << v[i].AsString().str() << std::endl;
+        // }
         auto &output = get_tensor(op.outputs(), 0);
         auto node = graph_.emplace<random_uniform>(to_data_type(output.type()), get_shape(output.shape()), 0.f, 1.f, time(nullptr));
         node->name(output.name()->string_view());