[lang] added templates for reductions

2019-09-10 15:54:16 -04:00
parent 41acac6ba1
commit 2781cdcf93
8 changed files with 124 additions and 15 deletions
--- a/include/triton/runtime/function.h
+++ b/include/triton/runtime/function.h
@@ -70,7 +70,7 @@ public:
  struct options_space_t {
    typedef std::pair<std::string, std::vector<std::string>> define_t;
    std::vector<define_t> defines;
-    std::vector<size_t> num_warps;
+    std::vector<int> num_warps;
  };

  struct options_t {