diff --git a/hog/cse.py b/hog/cse.py index bfc558e8499288fb813580d472b30a8a67e842a6..23d7c038a935ef63b5fb6d437baea0d0b21cd5ed 100644 --- a/hog/cse.py +++ b/hog/cse.py @@ -15,14 +15,12 @@ # along with this program. If not, see <https://www.gnu.org/licenses/>. import enum -from typing import Dict, Iterable, List, Tuple, Union +from typing import Dict, Iterable, List, Tuple import poly_cse_py.sympy_interface as poly_cse -import pystencils as ps import sympy as sp import hog.ast -from hog.exception import HOGException class CseImplementation(enum.Enum): diff --git a/hog/operator_generation/function_space_implementations/n1e1_space_impl.py b/hog/operator_generation/function_space_implementations/n1e1_space_impl.py index e9c48bf02d9a0f3d2794687254f011c529af026b..c2151357ce870379e135076c97b4591b87d23746 100644 --- a/hog/operator_generation/function_space_implementations/n1e1_space_impl.py +++ b/hog/operator_generation/function_space_implementations/n1e1_space_impl.py @@ -20,12 +20,7 @@ from typing import List, Tuple, Set, Union import pystencils as ps from pystencils import Field from pystencils.backend.ast import PsAstNode -from pystencils.backend.ast.expressions import ( - PsArrayInitList, - PsCast, - PsExpression, - PsLookup, -) +from pystencils.backend.ast.expressions import PsCast, PsExpression, PsLookup from pystencils.backend.ast.structural import PsDeclaration from pystencils.backend.functions import CFunction from pystencils.backend.kernelcreation import FreezeExpressions, KernelCreationContext diff --git a/hog/operator_generation/kernel_types.py b/hog/operator_generation/kernel_types.py index 5a6a601a6d816f29bef800b5b4332ffd3a96f8bd..28c151b59cfc44a0c205b291fc5efdce573d96d6 100644 --- a/hog/operator_generation/kernel_types.py +++ b/hog/operator_generation/kernel_types.py @@ -25,7 +25,6 @@ from sympy.codegen.ast import Assignment from pystencils import AssignmentCollection from pystencils.backend.ast import PsAstNode from pystencils.backend.ast.expressions import ( - PsArrayInitList, PsCast, PsExpression, PsLookup, diff --git a/hog/operator_generation/loop_strategies.py b/hog/operator_generation/loop_strategies.py index d300d28f8dcebb34756e5b873f194ecb45678d5f..cbb085d2a13679ba5df4c2619d9b960648f7fecc 100644 --- a/hog/operator_generation/loop_strategies.py +++ b/hog/operator_generation/loop_strategies.py @@ -28,7 +28,7 @@ involves balancing cache reuse and working set size. from abc import ABC, abstractmethod import sympy as sp -from typing import Dict, List, Type, Union +from typing import Dict, List, Union from pystencils.backend.constants import PsConstant from pystencils.backend.ast.expressions import PsExpression diff --git a/hog/operator_generation/pystencils_extensions.py b/hog/operator_generation/pystencils_extensions.py index f01d24fd3f6dbb800069ec2ea6e3de77e6e5f778..90e9c9186c6907210258d24a01b39a2f9ec188da 100644 --- a/hog/operator_generation/pystencils_extensions.py +++ b/hog/operator_generation/pystencils_extensions.py @@ -14,10 +14,9 @@ # You should have received a copy of the GNU General Public License # along with this program. If not, see <https://www.gnu.org/licenses/>. -from typing import Dict, List, Tuple, Union +from typing import Dict, List, Union from pystencils import Field, FieldType -from pystencils.backend.ast import PsAstNode from pystencils.backend.ast.structural import PsBlock, PsLoop from pystencils.backend.kernelcreation import AstFactory, KernelCreationContext from pystencils.defaults import DEFAULTS