Commit ba934fc2 authored by wsttiger's avatar wsttiger
Browse files

Formatting

parent 6ae81129
...@@ -54,7 +54,7 @@ std::vector<float> softmax(std::vector<float> p) ...@@ -54,7 +54,7 @@ std::vector<float> softmax(std::vector<float> p)
int main(int argc, char const* argv[]) int main(int argc, char const* argv[])
{ {
if (argc < 4) if(argc < 4)
{ {
throw std::runtime_error("Usage: cifar10 [gpu | cpu] <onnx file> <cifar10 data file>"); throw std::runtime_error("Usage: cifar10 [gpu | cpu] <onnx file> <cifar10 data file>");
} }
...@@ -65,7 +65,7 @@ int main(int argc, char const* argv[]) ...@@ -65,7 +65,7 @@ int main(int argc, char const* argv[])
std::cout << prog << std::endl; std::cout << prog << std::endl;
auto imageset = read_cifar10_images(datafile); auto imageset = read_cifar10_images(datafile);
if (gpu_cpu == "gpu") if(gpu_cpu == "gpu")
{ {
// GPU target // GPU target
prog.compile(migraph::gpu::target{}); prog.compile(migraph::gpu::target{});
......
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