diff --git a/src/engine/engine.cc b/src/engine/engine.cc index 9698b490157a..d5d8d3aa7a1b 100644 --- a/src/engine/engine.cc +++ b/src/engine/engine.cc @@ -21,12 +21,12 @@ inline Engine* CreateEngine() { ret = CreateNaiveEngine(); } else if (stype == "ThreadedEngine") { ret = CreateThreadedEnginePooled(); - } else if (stype == "ThreadedEnginePerDevie") { + } else if (stype == "ThreadedEnginePerDevice") { ret = CreateThreadedEnginePerDevice(); } CHECK_NE(ret, nullptr) - << "Cannot find Eine " << type << " in registry"; + << "Cannot find Engine " << type; if (!default_engine) { LOG(INFO) << "MXNet start using engine: " << type; } diff --git a/tests/cpp/threaded_engine_unittest.cc b/tests/cpp/threaded_engine_unittest.cc index 35e0ca3124b0..ffe3ee4ad3da 100644 --- a/tests/cpp/threaded_engine_unittest.cc +++ b/tests/cpp/threaded_engine_unittest.cc @@ -72,7 +72,7 @@ TEST(Engine, basics) { Foo(ctx, 42); cb(); }, - {}, {var}, mxnet::FnProperty::kCopy)); + {}, {var}, mxnet::FnProperty::kCopyFromGPU)); engine->Push(oprs.at(0), mxnet::Context{}); LOG(INFO) << "IO operator pushed, should wait for 2 seconds."; engine->WaitForVar(var);