Skip to content

Commit

Permalink
_dag_to_transformed_pytato_prg: use _DEFAULT_LOOPY_OPTIONS
Browse files Browse the repository at this point in the history
  • Loading branch information
matthiasdiener committed Jul 24, 2023
1 parent 7ce3ea8 commit d64e004
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions arraycontext/impl/pytato/compile.py
Original file line number Diff line number Diff line change
Expand Up @@ -391,7 +391,6 @@ def _dag_to_transformed_pytato_prg(self, dict_of_named_arrays, *, prg_id=None):
if prg_id is None:
prg_id = self.f

import loopy as lp
from pytato.target.loopy import BoundPyOpenCLProgram

self.actx._compile_trace_callback(
Expand All @@ -414,11 +413,10 @@ def _dag_to_transformed_pytato_prg(self, dict_of_named_arrays, *, prg_id=None):
prg_id, "pre_generate_loopy", pt_dict_of_named_arrays)

with ProcessLogger(logger, f"generate_loopy for '{prg_id}'"):
from arraycontext.loopy import _DEFAULT_LOOPY_OPTIONS
pytato_program = pt.generate_loopy(
pt_dict_of_named_arrays,
options=lp.Options(
return_dict=True,
no_numpy=True),
options=_DEFAULT_LOOPY_OPTIONS,
function_name=_prg_id_to_kernel_name(prg_id),
target=self.actx.get_target(),
)
Expand Down

0 comments on commit d64e004

Please sign in to comment.