Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Print the correct device type (#3478)
Summary: **Description** This commit modifies `resnet-runtime` to print the correct device type in its log messages. It currently prints `CPU` regardless of which backend is specified on the command line. **Testing** Ran the executable on the Interpreter and CPU and observed the log messages: ``` $ ../build_DEBUG/bin/resnet-runtime ../glow/tests/images/imagenet/ -backend=Interpreter WARNING: Logging before InitGoogleLogging() is written to STDERR I0830 07:35:04.535593 318879168 resnet-runtime.cpp:130] Initializing 5 Interpreter devices on HostManager. I0830 07:35:04.536674 318879168 resnet-runtime.cpp:78] Loading resnet50 model. $ ../build_DEBUG/bin/resnet-runtime ../glow/tests/images/imagenet/ -backend=CPU WARNING: Logging before InitGoogleLogging() is written to STDERR I0830 07:35:14.114750 161125824 resnet-runtime.cpp:130] Initializing 5 CPU devices on HostManager. I0830 07:35:14.115845 161125824 resnet-runtime.cpp:78] Loading resnet50 model. ``` **Fixes** This PR fixes #3471. Pull Request resolved: #3478 Reviewed By: rdzhabarov Differential Revision: D17132626 Pulled By: SplitInfinity fbshipit-source-id: 238189d0f50e15a43acc45f659970cf70e19ca2c
- Loading branch information