You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

include.patch 1.7 kB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455
  1. diff -Npur tvm/include/tvm/expr_operator.h tvm_new/include/tvm/expr_operator.h
  2. --- tvm/include/tvm/expr_operator.h 2019-12-28 10:11:27.369814744 +0800
  3. +++ tvm_new/include/tvm/expr_operator.h 2019-12-28 10:11:27.209812391 +0800
  4. @@ -25,6 +25,11 @@
  5. * when the type is int32 or int64 for simplifying the index expressions.
  6. */
  7. // Acknowledgement: Most operator APIs originate from Halide.
  8. +
  9. +/*
  10. + * 2019.12.30 - Add new operator for expr.
  11. + */
  12. +
  13. #ifndef TVM_EXPR_OPERATOR_H_
  14. #define TVM_EXPR_OPERATOR_H_
  15. @@ -217,6 +222,16 @@ TVM_DLL Expr operator*(Expr a, Expr b);
  16. */
  17. TVM_DLL Expr operator/(Expr a, Expr b);
  18. /*!
  19. + * \brief mod operator
  20. + *
  21. + * \param a left operand
  22. + * \param b right operand
  23. + * \return The result expression.
  24. + * \note this function does eager constant folding for
  25. + * index types(int32, int64) when possible.
  26. + */
  27. +TVM_DLL Expr operator%(Expr a, Expr b);
  28. +/*!
  29. * \brief left shift operator
  30. *
  31. * \param a left operand
  32. diff -Npur tvm/include/tvm/lowered_func.h tvm_new/include/tvm/lowered_func.h
  33. --- tvm/include/tvm/lowered_func.h 2019-12-28 10:11:27.369814744 +0800
  34. +++ tvm_new/include/tvm/lowered_func.h 2019-12-28 10:11:27.209812391 +0800
  35. @@ -22,6 +22,11 @@
  36. * \brief Information about a lowered TVM function.
  37. * This data structure is final step toward codegen.
  38. */
  39. +
  40. +/*
  41. + * 2019.12.30 - Add new var array for args_real.
  42. + */
  43. +
  44. #ifndef TVM_LOWERED_FUNC_H_
  45. #define TVM_LOWERED_FUNC_H_
  46. @@ -74,6 +79,7 @@ class LoweredFuncNode : public ir::Funct
  47. * This function can only take pod type(int, float) and void* as arguments.
  48. */
  49. Array<Var> args;
  50. + Array<Var> args_real;
  51. /*!
  52. * \brief The IterVar axis of threads
  53. * Each axis need host function to specify a size.