Skip to content

Commit

Permalink
Fix plint error.
Browse files Browse the repository at this point in the history
  • Loading branch information
huajsj committed May 27, 2021
1 parent b6fbf7d commit 445488c
Show file tree
Hide file tree
Showing 3 changed files with 7 additions and 6 deletions.
3 changes: 2 additions & 1 deletion src/runtime/pipeline/pipeline_executor.cc
Original file line number Diff line number Diff line change
Expand Up @@ -18,9 +18,10 @@
*/

/*!
* \file pipeline_runtime.cc
* \file pipeline_executor.cc
*/
#include "pipeline_executor.h"

#include <tvm/runtime/registry.h>

namespace tvm {
Expand Down
6 changes: 3 additions & 3 deletions src/runtime/pipeline/pipeline_executor.h
Original file line number Diff line number Diff line change
Expand Up @@ -18,9 +18,8 @@
*/

/*!
* \brief Tiny graph runtime that can run graph
* containing only tvm PackedFunc.
* \file graph_runtime.h
* \brief pipeline executor
* \file pipeline_executor.h
*/
#ifndef TVM_RUNTIME_PIPELINE_PIPELINE_EXECUTOR_H_
#define TVM_RUNTIME_PIPELINE_PIPELINE_EXECUTOR_H_
Expand All @@ -31,6 +30,7 @@

#include "../file_utils.h"
#include "pipeline_function.h"

using namespace std;
namespace tvm {
namespace runtime {
Expand Down
4 changes: 2 additions & 2 deletions src/runtime/pipeline/pipeline_struct.h
Original file line number Diff line number Diff line change
Expand Up @@ -36,6 +36,7 @@
#include <vector>
#define SLOT slot_t<>
#define SUB_Q_SIZE 1024

using namespace tvm::runtime;
using namespace std;
typedef unordered_map<int, unordered_map<int, int>> RUNTIME_PIPELINE_OUTPUT_CONF;
Expand Down Expand Up @@ -379,8 +380,7 @@ class RuntimeFunction {
*/
void CopyFromTo(DLTensor* from, DLTensor* to) {
if (!(from->device.device_type == to->device.device_type ||
from->device.device_type == kDLCPU || to->device.device_type == kDLCPU ||
from->device.device_type == kDLCPUPinned || to->device.device_type == kDLCPUPinned)) {
from->device.device_type == kDLCPU || to->device.device_type == kDLCPU)) {
if (dlLocal == nullptr) {
dlLocal = CreateFromDLTensor(from);
}
Expand Down

0 comments on commit 445488c

Please sign in to comment.