[python] more progress towards tensorflow/pytorch unification

2019-09-04 12:47:59 -04:00
parent cdbc9d4ecd
commit f6e9c24fe8
3 changed files with 132 additions and 104 deletions
--- a/lib/runtime/function.cc
+++ b/lib/runtime/function.cc
@@ -148,8 +148,9 @@ function::caller function::autotune(driver::stream* stream, const grid_fn_ty& gr
    options_t opt;
    unsigned i = 0;
    opt.num_warps = std::stoi(params[i++]);
-    for(auto it: opt_space_.defines)
+    for(auto it: opt_space_.defines){
      opt.defines[it.first] = params[i++];
+    }
    // pre-process
    TokenSequence tokens;
    Preprocessor cpp(&src_, true);