Merge pull request #2 from arhik/main

missing `where` args

Author: arhik

src/codegen/computeBlock.jl

@@ -37,7 +37,8 @@ function computeBlock(scope, islaunch, wgSize, wgCount, fname, fargs)
 	fn = inferExpr(childScope, fname)
 	fa = map(x -> inferExpr(childScope, x), fargs)
 	fb = map(x -> inferExpr(childScope, x), fbody)
-	return ComputeBlock(fn, fa, WGPUVariable[], fb, childScope)
+	ta = map(x -> inferExpr(childScope, x), Targs)
+	return ComputeBlock(fn, fa, ta, fb, childScope)
 end
 
 

src/codegen/transpile.jl

@@ -79,5 +79,6 @@ function transpile(scope::Scope, computeBlk::ComputeBlock)
 	fn = transpile(scope::Scope, computeBlk.fname)
 	fa = map(x -> transpile(scope, x), computeBlk.fargs)
 	fb = map(x -> transpile(scope, x), computeBlk.fbody)
-	return Expr(:function, Expr(:call, fn, fa...), quote $(fb...) end) |> MacroTools.striplines
+	ta = map(x -> transpile(scope, x), computeBlk.Targs)
+	return Expr(:function, Expr(:where, Expr(:call, fn, fa...), ta...), quote $(fb...) end) |> MacroTools.striplines
 end