|
|
|
@@ -736,11 +736,11 @@ std::string ExtractGraphKernelName(const AnfNodePtrList &cnodes, const string &p |
|
|
|
std::vector<PrimitivePtr> GetFusibleOpList() { |
|
|
|
#if ENABLE_D |
|
|
|
std::vector<PrimitivePtr> fusible_basic_ops = { |
|
|
|
prim::kPrimAbs, prim::kPrimRound, prim::kPrimNeg, prim::kPrimExp, prim::kPrimTensorAdd, |
|
|
|
prim::kPrimExpandDims, prim::kPrimMul, prim::kPrimMinimum, prim::kPrimMaximum, prim::kPrimLog, |
|
|
|
prim::kPrimPow, prim::kPrimSub, prim::kPrimRsqrt, prim::kPrimSqrt, prim::kPrimAddN, |
|
|
|
prim::kPrimEqual, prim::kPrimReciprocal, prim::kPrimReduceSum, prim::kPrimTanh, prim::kPrimReshape, |
|
|
|
prim::kPrimTranspose, prim::kPrimCast, prim::kPrimRealDiv}; |
|
|
|
prim::kPrimAbs, prim::kPrimRound, prim::kPrimNeg, prim::kPrimExp, prim::kPrimTensorAdd, |
|
|
|
prim::kPrimExpandDims, prim::kPrimMul, prim::kPrimMinimum, prim::kPrimMaximum, prim::kPrimLog, |
|
|
|
prim::kPrimPow, prim::kPrimSub, prim::kPrimRsqrt, prim::kPrimSqrt, prim::kPrimAddN, |
|
|
|
prim::kPrimEqual, prim::kPrimReciprocal, prim::kPrimTanh, prim::kPrimReshape, prim::kPrimTranspose, |
|
|
|
prim::kPrimCast, prim::kPrimRealDiv}; |
|
|
|
#elif ENABLE_GPU |
|
|
|
std::vector<PrimitivePtr> fusible_basic_ops = { |
|
|
|
prim::kPrimAbs, prim::kPrimRound, prim::kPrimNeg, prim::kPrimExp, prim::kPrimTensorAdd, |
|
|
|
|