move uniform and localmem outside control-flow

Author: vchuravy

src/macros.jl

@@ -105,6 +105,15 @@ function transform_gpu!(def, constargs, force_inbounds)
             push!(new_stmts, stmt)
             continue
         end
+        if @capture(stmt, @uniform x_)
+            push!(new_stmts, stmt)
+            continue
+        elseif @capture(stmt, lhs_ = rhs_ | (vs__, lhs_ = rhs_))
+            if @capture(rhs, @localmem(args__) | @uniform(args__))
+                push!(new_stmts, stmt)
+                continue
+            end
+        end
         push!(active_stmts, stmt)
     end
     push!(new_stmts, Expr(:if, :__active_lane__, Expr(:block, active_stmts...)))