/** * \file imperative/src/impl/ops/tensorrt_runtime.cpp * MegEngine is Licensed under the Apache License, Version 2.0 (the "License") * * Copyright (c) 2014-2021 Megvii Inc. All rights reserved. * * Unless required by applicable law or agreed to in writing, * software distributed under the License is distributed on an * "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or * implied. */#include "../op_trait.h"#include "megbrain/imperative/ops/autogen.h"#if MGB_ATLAS#include "megbrain/opr/atlas_runtime_op.h"namespacemgb::imperative{namespace{namespaceatlas_runtime{autoapply_on_var_node(constOpDef&def,constVarNodeArray&inputs){auto&&op=static_cast<constAtlasRuntime&>(def);SymbolVarArraysymbol_var_inputs(inputs.begin(),inputs.end());OperatorNodeConfigconfig{op.make_name()};returnopr::AtlasRuntimeOpr::make(op.buf.c_str(),op.buf_size,symbol_var_inputs,config);}OP_TRAIT_REG(AtlasRuntime,AtlasRuntime).apply_on_var_node(apply_on_var_node).fallback();}// namespace atlas_runtime}// namespace}// namespace mgb::imperative#endif