Skip to content

Commit 9b9395e

Browse files
committed
simplify
1 parent e6c2c0a commit 9b9395e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/ProbProg.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@ function generate(f::Function, args::Vararg{Any,Nargs}) where {Nargs}
1111
resargprefix::Symbol = gensym("generateresarg")
1212

1313
mlir_fn_res = TracedUtils.make_mlir_fn(
14-
f, args, (), string(f) * "_generate", false; argprefix, resprefix, resargprefix
14+
f, args, (), string(f), false; argprefix, resprefix, resargprefix
1515
)
1616
(; result, linear_args, in_tys, linear_results) = mlir_fn_res
1717
fnwrap = mlir_fn_res.fnwrapped
@@ -69,7 +69,7 @@ function sample(f::Function, args::Vararg{Any,Nargs}) where {Nargs}
6969
resargprefix = gensym("sampleresarg")
7070

7171
mlir_fn_res = TracedUtils.make_mlir_fn(
72-
f, args, (), string(f) * "_sample", false; argprefix, resprefix, resargprefix
72+
f, args, (), string(f), false; argprefix, resprefix, resargprefix
7373
)
7474
(; result, linear_args, in_tys, linear_results) = mlir_fn_res
7575
fnwrap = mlir_fn_res.fnwrapped

0 commit comments

Comments
 (0)