Skip to content

Commit 92213e3

Browse files
committed
formatting + removal of unnecessary argument
1 parent 0220d5c commit 92213e3

2 files changed

Lines changed: 7 additions & 5 deletions

File tree

pykokkos/core/cpp_setup.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -222,7 +222,9 @@ def generate_cmake(
222222
view_layout_space = default_space if not is_host_execution_space(default_space) else space
223223
else:
224224
view_layout_space = space
225-
view_layout: str = str(get_default_layout(get_default_memory_space(view_layout_space)))
225+
view_layout: str = str(
226+
get_default_layout(get_default_memory_space(view_layout_space))
227+
)
226228
view_layout = view_layout.split(".")[-1]
227229
view_layout = f"Kokkos::{view_layout}"
228230

pykokkos/core/translators/bindings.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -142,7 +142,6 @@ def get_device_views(members: PyKokkosMembers) -> Dict[str, str]:
142142
}
143143

144144

145-
146145
def _generate_mirror_with_exec_layout(
147146
src: str,
148147
dst: str,
@@ -167,12 +166,13 @@ def _generate_mirror_with_exec_layout(
167166
extents: str = ",".join(f"{src}.extent({i})" for i in range(rank))
168167

169168
code: str = (
170-
f'{dst_type} {dst}('
169+
f"{dst_type} {dst}("
171170
f'Kokkos::view_alloc("{dst}", Kokkos::WithoutInitializing), {extents});'
172171
f"Kokkos::deep_copy({dst}, {src});"
173172
)
174173
return code
175174

175+
176176
def generate_functor_instance(
177177
functor: str,
178178
members: PyKokkosMembers,
@@ -215,13 +215,13 @@ def generate_functor_instance(
215215
== Keywords.ArgMemSpace.value
216216
):
217217
mirror_views += _generate_mirror_with_exec_layout(
218-
v, d_v, view_type, exec_space_instance, None
218+
v, d_v, view_type, None
219219
)
220220
else:
221221
mirror_views += f"auto {d_v} = {v};"
222222
else:
223223
mirror_views += _generate_mirror_with_exec_layout(
224-
v, d_v, members.views[cppast.DeclRefExpr(v)], exec_space_instance, None
224+
v, d_v, members.views[cppast.DeclRefExpr(v)], None
225225
)
226226

227227
# Kokkos fails to compile a functor if there are no parameters in its constructor

0 commit comments

Comments
 (0)