// Copyright 2019 Fuzhou Rockchip Electronics Co., Ltd. All rights reserved.
|
// Use of this source code is governed by a BSD-style license that can be
|
// found in the LICENSE file.
|
|
#include "ai_task_factory.h"
|
#include <assert.h>
|
#include "logger/log.h"
|
#ifdef HAVE_STASTERIA
|
#include "st_task_handler.h"
|
#elif HAVE_ROCKX
|
#include "rockx_task_handler.h"
|
#endif
|
|
namespace rockchip {
|
namespace aiserver {
|
|
AITaskHandler* AITaskFactory::createHandler() {
|
AITaskHandler *handler = nullptr;
|
#ifdef HAVE_STASTERIA
|
handler = new STTaskHandler();
|
#elif HAVE_ROCKX
|
handler = new RockxTaskHandler();
|
#else
|
LOG_ERROR("Error configuration for NN handler");
|
assert(0);
|
#endif
|
|
return handler;
|
}
|
|
} // namespace aiserver
|
} // namespace rockchip
|
