Moved resnet18 to cifar10 executable

src/onnx/CMakeLists.txt

@@ -21,9 +21,9 @@ add_executable(mnist mnist.cpp)
 rocm_clang_tidy_check(mnist)
 target_link_libraries(mnist migraph_cpu migraph_gpu migraph_onnx)
 
-add_executable(resnet18 resnet18.cpp)
-rocm_clang_tidy_check(resnet18)
-target_link_libraries(resnet18 migraph_cpu migraph_gpu migraph_onnx)
+add_executable(cifar10 cifar10.cpp)
+rocm_clang_tidy_check(cifar10)
+target_link_libraries(cifar10 migraph_cpu migraph_gpu migraph_onnx)
 
 if(MIGRAPH_ENABLE_GPU)
 add_executable(verify_onnx verify_onnx.cpp)

src/onnx/resnet18.cpp → src/onnx/cifar10.cpp

@@ -11,9 +11,8 @@
 #include <migraph/gpu/hip.hpp>
 #include <migraph/generate.hpp>
 
-auto read_cifar10_images(std::string full_path)
+auto read_cifar10_images(const std::string& full_path)
 {
-
     std::ifstream file(full_path, std::ios::binary);
 
     const size_t nimages          = 10;
@@ -55,40 +54,46 @@ std::vector<float> softmax(std::vector<float> p)
 
 int main(int argc, char const* argv[])
 {
-    std::string file     = argv[1];
-    std::string datafile = argv[2];
+    if (argc < 4)
+    {
+        throw std::runtime_error("Usage:  cifar10 [gpu | cpu] <onnx file> <cifar10 data file>");
+    }
+    std::string gpu_cpu  = argv[1];
+    std::string file     = argv[2];
+    std::string datafile = argv[3];
     auto prog            = migraph::parse_onnx(file);
     std::cout << prog << std::endl;
     auto imageset = read_cifar10_images(datafile);
 
-    // // GPU target
-    // prog.compile(migraph::gpu::target{});
-    // migraph::program::parameter_map m;
-    // auto s = migraph::shape{migraph::shape::float_type, {1, 3, 32, 32}};
-    // for(auto&& x : prog.get_parameter_shapes())
-    // {
-    //     m[x.first] = migraph::gpu::to_gpu(migraph::generate_argument(x.second));
-    // }
-    // auto labels = imageset.first;
-    // auto input  = imageset.second;
-    // auto ptr    = input.data();
-    // for(int i = 0; i < 10; i++)
-    // {
-    //     std::cout << "label: " << (uint32_t)labels[i] << "  ---->  ";
-    //     m["0"]      = migraph::gpu::to_gpu(migraph::argument{s, &ptr[3072 * i]});
-    //     auto result = migraph::gpu::from_gpu(prog.eval(m));
-    //     std::vector<float> logits;
-    //     result.visit([&](auto output) { logits.assign(output.begin(), output.end()); });
-    //     std::vector<float> probs = softmax(logits);
-    //     for(auto x : logits)
-    //         //std::cout << x << "  ";
-    //         //std::cout << x << std::endl;
-    //         printf("%10.5e    ", x);
-    //     std::cout << std::endl;
-    //     std::cout << std::endl;
-    // }
-
-    // // CPU target
+    if (gpu_cpu == "gpu")
+    {
+        // GPU target
+        prog.compile(migraph::gpu::target{});
+        migraph::program::parameter_map m;
+        auto s = migraph::shape{migraph::shape::float_type, {1, 3, 32, 32}};
+        for(auto&& x : prog.get_parameter_shapes())
+        {
+            m[x.first] = migraph::gpu::to_gpu(migraph::generate_argument(x.second));
+        }
+        auto labels = imageset.first;
+        auto input  = imageset.second;
+        auto ptr    = input.data();
+        for(int i = 0; i < 10; i++)
+        {
+            std::cout << "label: " << static_cast<uint32_t>(labels[i]) << "  ---->  ";
+            m["0"]      = migraph::gpu::to_gpu(migraph::argument{s, &ptr[3072 * i]});
+            auto result = migraph::gpu::from_gpu(prog.eval(m));
+            std::vector<float> logits;
+            result.visit([&](auto output) { logits.assign(output.begin(), output.end()); });
+            std::vector<float> probs = softmax(logits);
+            for(auto x : probs)
+                std::cout << x << "    ";
+            std::cout << std::endl << std::endl;
+        }
+    }
+    else
+    {
+        // CPU target
         prog.compile(migraph::cpu::cpu_target{});
         auto s      = migraph::shape{migraph::shape::float_type, {1, 3, 32, 32}};
         auto labels = imageset.first;
@@ -96,14 +101,15 @@ int main(int argc, char const* argv[])
         auto ptr    = input.data();
         for(int i = 0; i < 10; i++)
         {
-        std::cout << "label: " << (uint32_t)labels[i] << "  ---->  ";
+            std::cout << "label: " << static_cast<uint32_t>(labels[i]) << "  ---->  ";
             auto input3 = migraph::argument{s, &ptr[3072 * i]};
             auto result = prog.eval({{"0", input3}});
             std::vector<float> logits;
             result.visit([&](auto output) { logits.assign(output.begin(), output.end()); });
             std::vector<float> probs = softmax(logits);
-        for(auto x : logits)
-            printf("%10.5e    ", x);
+            for(auto x : probs)
+                std::cout << x << "    ";
             std::cout << std::endl;
         }
+    }
 }

src/program.cpp

@@ -292,7 +292,6 @@ argument generic_eval(const program& p,
         }
         else if(ins->op.name() == "@param")
         {
-            std::cout << ins->op.name() << std::endl;
             results.emplace(ins, trace(ins, [&] {
                                 return params.at(any_cast<builtin::param>(ins->op).parameter);
                             }));