Skip to content

Commit

Permalink
Use make_kernel to force entrypoint registration
Browse files Browse the repository at this point in the history
---------

Co-authored-by: Jack Betteridge <43041811+JDBetteridge@users.noreply.github.com>
  • Loading branch information
connorjward and JDBetteridge authored Oct 22, 2024
1 parent 639b86a commit 8d9abb0
Showing 1 changed file with 11 additions and 3 deletions.
14 changes: 11 additions & 3 deletions tsfc/loopy.py
Original file line number Diff line number Diff line change
Expand Up @@ -246,9 +246,17 @@ def generate(impero_c, args, scalar_type, kernel_name="loopy_kernel", index_name
domains = create_domains(ctx.index_extent.items())

# Create loopy kernel
knl = lp.make_function(domains, instructions, data, name=kernel_name, target=target,
seq_dependencies=True, silenced_warnings=["summing_if_branches_ops"],
lang_version=(2018, 2), preambles=preamble)
knl = lp.make_kernel(
domains,
instructions,
data,
name=kernel_name,
target=target,
seq_dependencies=True,
silenced_warnings=["summing_if_branches_ops"],
lang_version=(2018, 2),
preambles=preamble
)

# Prevent loopy interchange by loopy
knl = lp.prioritize_loops(knl, ",".join(ctx.index_extent.keys()))
Expand Down

0 comments on commit 8d9abb0

Please sign in to comment.