Browse Source

fix(imperative/ops): use tblgen to generate FastpathCopy

GitOrigin-RevId: b02157dacb
release-1.5
Megvii Engine Team 3 years ago
parent
commit
5f558042b2
4 changed files with 3 additions and 16 deletions
  1. +1
    -0
      imperative/python/src/tensor.cpp
  2. +0
    -2
      imperative/src/impl/ops/utility.cpp
  3. +0
    -14
      imperative/src/include/megbrain/imperative/ops/utility.h
  4. +2
    -0
      src/core/include/megbrain/ir/ops.td

+ 1
- 0
imperative/python/src/tensor.cpp View File

@@ -13,6 +13,7 @@
#include "megbrain/common.h" #include "megbrain/common.h"
#include "megbrain/imperative/ops/utility.h" #include "megbrain/imperative/ops/utility.h"
#include "megbrain/imperative/ops/backward_graph.h" #include "megbrain/imperative/ops/backward_graph.h"
#include "megbrain/imperative/ops/autogen.h"
#include "megbrain/imperative/profiler.h" #include "megbrain/imperative/profiler.h"
#include "megbrain/opr/io.h" #include "megbrain/opr/io.h"




+ 0
- 2
imperative/src/impl/ops/utility.cpp View File

@@ -31,8 +31,6 @@ OP_TRAIT_REG(FastpathCopy,FastpathCopy)
.fallback(); .fallback();
}} // fastpathcopy }} // fastpathcopy


MGB_DYN_TYPE_OBJ_FINAL_IMPL(FastpathCopy);

namespace { namespace identity { namespace { namespace identity {
auto apply_on_var_node( auto apply_on_var_node(
const OpDef& def, const OpDef& def,


+ 0
- 14
imperative/src/include/megbrain/imperative/ops/utility.h View File

@@ -35,18 +35,4 @@ struct GenericPyOp final : OpDefImplBase<GenericPyOp> {
MGB_DYN_TYPE_OBJ_FINAL_DECL; MGB_DYN_TYPE_OBJ_FINAL_DECL;
}; };


struct FastpathCopy final : OpDefImplBase<FastpathCopy> {
FastpathCopy() = default;

size_t hash() const override {
return mgb::hash(this->dyn_typeinfo());
}

bool is_same_st(const Hashable& rhs) const override {
return this->dyn_typeinfo() == rhs.dyn_typeinfo();
}

MGB_DYN_TYPE_OBJ_FINAL_DECL;
};

} // namespace mgb::imperative } // namespace mgb::imperative

+ 2
- 0
src/core/include/megbrain/ir/ops.td View File

@@ -375,4 +375,6 @@ def CvtColor: MgbHashableOp<"CvtColor", [CvtColorParam]>;


def CheckHasInf: MgbHashableOp<"CheckHasInf", [EmptyParam]>; def CheckHasInf: MgbHashableOp<"CheckHasInf", [EmptyParam]>;


def FastpathCopy: MgbHashableOp<"FastpathCopy">;

#endif // MGB_OPS #endif // MGB_OPS

Loading…
Cancel
Save