Translate rand! and randn! to mmap! instead of mmap

IntelLabs · Dec 28, 2015 · b046a90 · b046a90
1 parent c0ced45
commit b046a90
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/domain-ir.jl b/src/domain-ir.jl
@@ -1298,7 +1298,8 @@ function translate_call_map(state, env, typ, fun, args::Array{Any,1})
         end
         args[i] = arg_
     end
-    expr = mmapRemoveDupArg!(mk_mmap(args, domF))
+    expr = endswith(string(fun), '!') ? mk_mmap(args, domF) : mk_mmap!(args, domF)
+    expr = mmapRemoveDupArg!(expr)
     expr.typ = typ
     return expr
 end