Skip to content

Commit

Permalink
fix lint
Browse files Browse the repository at this point in the history
  • Loading branch information
vvchernov committed Feb 21, 2022
1 parent 289bd82 commit 4e01e40
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 15 deletions.
19 changes: 7 additions & 12 deletions include/tvm/runtime/vm/vm.h
Original file line number Diff line number Diff line change
Expand Up @@ -302,25 +302,20 @@ class VirtualMachine : public runtime::ModuleNode {
* \param func_name The function's name.
* \return The input tensor index.
*/
int64_t getInputIndexFromName(const std::string& input_name,
const std::string& func_name) const;
int64_t getInputIndexFromName(const std::string& input_name, const std::string& func_name) const;
/*!
* \brief Check executable exists and function name is in global map, get VM function.
* \param func_name The function's name.
* \return VM function.
*/
const VMFunction& checkAndGetVMFunction(const std::string& func_name) const;
/*!
* \brief Set one input tensor with given index to set of input tensors if need copy to given device.
* \param tensors the input tensors set (destination)
* \param tensor some tensor (not neccessary DLTensor)
* \param index The input tensor index.
* \param dev device to copy if need.
/*!
* \brief Set one input tensor with given index to set of input tensors if need copy to given
* device. \param tensors the input tensors set (destination) \param tensor some tensor (not
* neccessary DLTensor). \param index The input tensor index. \param dev device to copy if need.
*/
void SetInputTensorWithIndex(std::vector<ObjectRef>& tensors, // NOLINT(*)
const TVMArgValue& tensor,
int index,
Device dev);
void SetInputTensorWithIndex(std::vector<ObjectRef>& tensors, // NOLINT(*)
const TVMArgValue& tensor, int index, Device dev);

protected:
/*! \brief The virtual machine's packed function table. */
Expand Down
4 changes: 1 addition & 3 deletions src/runtime/vm/vm.cc
Original file line number Diff line number Diff line change
Expand Up @@ -288,9 +288,7 @@ const VMFunction& VirtualMachine::checkAndGetVMFunction(const std::string& func_
}

void VirtualMachine::SetInputTensorWithIndex(std::vector<ObjectRef>& tensors,
const TVMArgValue& inp_tensor,
int index,
Device dev) {
const TVMArgValue& inp_tensor, int index, Device dev) {
if (inp_tensor.type_code() == kTVMDLTensorHandle) {
// Automatically convert input DLTensors to NDArray
DLTensor* tensor = inp_tensor;
Expand Down

0 comments on commit 4e01e40

Please sign in to comment.