Skip to content
Snippets Groups Projects
Commit 0fd6f004 authored by Jan Hönig's avatar Jan Hönig
Browse files

Removed unnecessary imports.

parent edc023ad
No related branches found
No related tags found
1 merge request!292Rebase of pystencils Type System
from typing import Union from typing import Union
import sympy as sp
import numpy as np import numpy as np
import pystencils.astnodes as ast import pystencils.astnodes as ast
...@@ -9,9 +8,9 @@ from pystencils.config import CreateKernelConfig ...@@ -9,9 +8,9 @@ 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 from pystencils.typing import StructType, TypedSymbol
from pystencils.typing.transformations import add_types from pystencils.typing.transformations import add_types
from pystencils.field import Field, FieldType from pystencils.field import FieldType
from pystencils.node_collection import NodeCollection from pystencils.node_collection import NodeCollection
from pystencils.transformations import ( from pystencils.transformations import (
filtered_tree_iteration, get_base_buffer_index, get_optimal_loop_ordering, make_loop_over_domain, filtered_tree_iteration, get_base_buffer_index, get_optimal_loop_ordering, make_loop_over_domain,
...@@ -46,7 +45,6 @@ def create_kernel(assignments: Union[AssignmentCollection, NodeCollection], ...@@ -46,7 +45,6 @@ def create_kernel(assignments: Union[AssignmentCollection, NodeCollection],
assignments = assignments.all_assignments assignments = assignments.all_assignments
# TODO Jan Cleanup: move add_types to create_domain_kernel or create_kernel? # TODO Jan Cleanup: move add_types to create_domain_kernel or create_kernel?
assignments = add_types(assignments, config) assignments = add_types(assignments, config)
all_fields = fields_read.union(fields_written) all_fields = fields_read.union(fields_written)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment