Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
Frederik Hennig
pystencils
Commits
ea943334
Commit
ea943334
authored
May 11, 2022
by
Markus Holzer
Browse files
Fix flake8
parent
1386091a
Changes
2
Hide whitespace changes
Inline
Side-by-side
pystencils/boundaries/createindexlist.py
View file @
ea943334
...
...
@@ -46,8 +46,8 @@ def _create_index_list_python(flag_field_arr, boundary_mask,
nr_of_ghost_layers
=
0
coordinate_names
=
boundary_index_array_coordinate_names
[:
len
(
flag_field_arr
.
shape
)]
index_arr_dtype
=
np
.
dtype
([(
name
,
default_index_array_dtype
)
for
name
in
coordinate_names
]
+
[(
direction_member_name
,
default_index_array_dtype
)])
index_arr_dtype
=
np
.
dtype
([(
name
,
default_index_array_dtype
)
for
name
in
coordinate_names
]
+
[(
direction_member_name
,
default_index_array_dtype
)])
# boundary cells are extracted via np.where. To ensure continous memory access in the compute kernel these cells
# have to be sorted.
...
...
@@ -119,8 +119,8 @@ def create_boundary_index_list(flag_field, stencil, boundary_mask, fluid_mask,
"""
dim
=
len
(
flag_field
.
shape
)
coordinate_names
=
boundary_index_array_coordinate_names
[:
dim
]
index_arr_dtype
=
np
.
dtype
([(
name
,
default_index_array_dtype
)
for
name
in
coordinate_names
]
+
[(
direction_member_name
,
default_index_array_dtype
)])
index_arr_dtype
=
np
.
dtype
([(
name
,
default_index_array_dtype
)
for
name
in
coordinate_names
]
+
[(
direction_member_name
,
default_index_array_dtype
)])
stencil
=
np
.
array
(
stencil
,
dtype
=
default_index_array_dtype
)
args
=
(
flag_field
,
nr_of_ghost_layers
,
boundary_mask
,
fluid_mask
,
stencil
,
single_link
)
...
...
pystencils/cpu/kernelcreation.py
View file @
ea943334
from
typing
import
Union
import
sympy
as
sp
import
numpy
as
np
import
pystencils.astnodes
as
ast
from
pystencils.simp.assignment_collection
import
AssignmentCollection
...
...
@@ -9,7 +8,7 @@ from pystencils.config import CreateKernelConfig
from
pystencils.enums
import
Target
,
Backend
from
pystencils.astnodes
import
Block
,
KernelFunction
,
LoopOverCoordinate
,
SympyAssignment
from
pystencils.cpu.cpujit
import
make_python_function
from
pystencils.typing
import
StructType
,
TypedSymbol
,
create_type
,
get_type_of_expression
from
pystencils.typing
import
StructType
,
TypedSymbol
,
create_type
from
pystencils.typing.transformations
import
add_types
from
pystencils.field
import
Field
,
FieldType
from
pystencils.node_collection
import
NodeCollection
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment