Skip to content
Snippets Groups Projects
Commit dd477a5a authored by Dominic Kempf's avatar Dominic Kempf
Browse files

fixup 2

parent e4467b6f
No related branches found
No related tags found
No related merge requests found
......@@ -68,6 +68,11 @@ def _realize_input(sf, insn_dep):
context_tags=("kernel",),
cache_key_generator=lambda s, **kw: s.cache_key)
def _realize_sum_factorization_kernel(sf, insn_dep=frozenset(), outshape=None, direct_input=None, direct_output=None):
# Unify the insn_dep parameter to be a frozenset
if isinstance(insn_dep, str):
insn_dep = frozenset({insn_dep})
assert isinstance(insn_dep, frozenset)
# Get the vectorization info. During dry run, this is a now op
# sf = attach_vectorization_info(sf)
if get_global_context_value("dry_run", False):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment