Skip to content
Snippets Groups Projects
Commit ea943334 authored by Markus Holzer's avatar Markus Holzer
Browse files

Fix flake8

parent 1386091a
1 merge request!292Rebase of pystencils Type System
...@@ -46,8 +46,8 @@ def _create_index_list_python(flag_field_arr, boundary_mask, ...@@ -46,8 +46,8 @@ def _create_index_list_python(flag_field_arr, boundary_mask,
nr_of_ghost_layers = 0 nr_of_ghost_layers = 0
coordinate_names = boundary_index_array_coordinate_names[:len(flag_field_arr.shape)] 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] + index_arr_dtype = np.dtype([(name, default_index_array_dtype) for name in coordinate_names]
[(direction_member_name, default_index_array_dtype)]) + [(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 # boundary cells are extracted via np.where. To ensure continous memory access in the compute kernel these cells
# have to be sorted. # have to be sorted.
...@@ -119,8 +119,8 @@ def create_boundary_index_list(flag_field, stencil, boundary_mask, fluid_mask, ...@@ -119,8 +119,8 @@ def create_boundary_index_list(flag_field, stencil, boundary_mask, fluid_mask,
""" """
dim = len(flag_field.shape) dim = len(flag_field.shape)
coordinate_names = boundary_index_array_coordinate_names[:dim] coordinate_names = boundary_index_array_coordinate_names[:dim]
index_arr_dtype = np.dtype([(name, default_index_array_dtype) for name in coordinate_names] + index_arr_dtype = np.dtype([(name, default_index_array_dtype) for name in coordinate_names]
[(direction_member_name, default_index_array_dtype)]) + [(direction_member_name, default_index_array_dtype)])
stencil = np.array(stencil, dtype=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) args = (flag_field, nr_of_ghost_layers, boundary_mask, fluid_mask, stencil, single_link)
......
from typing import Union from typing import Union
import sympy as sp import sympy as sp
import numpy as np
import pystencils.astnodes as ast import pystencils.astnodes as ast
from pystencils.simp.assignment_collection import AssignmentCollection from pystencils.simp.assignment_collection import AssignmentCollection
...@@ -9,7 +8,7 @@ from pystencils.config import CreateKernelConfig ...@@ -9,7 +8,7 @@ from pystencils.config import CreateKernelConfig
from pystencils.enums import Target, Backend from pystencils.enums import Target, Backend
from pystencils.astnodes import Block, KernelFunction, LoopOverCoordinate, SympyAssignment from pystencils.astnodes import Block, KernelFunction, LoopOverCoordinate, SympyAssignment
from pystencils.cpu.cpujit import make_python_function 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.typing.transformations import add_types
from pystencils.field import Field, FieldType from pystencils.field import Field, FieldType
from pystencils.node_collection import NodeCollection from pystencils.node_collection import NodeCollection
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment