diff --git a/src/pystencils/backends/arm_instruction_sets.py b/src/pystencils/backends/arm_instruction_sets.py index 3e50d5f45374b2ece2aa21487db82a8e8ebb0097..92494205ab839548b5dc823b0b53f70da8530ab9 100644 --- a/src/pystencils/backends/arm_instruction_sets.py +++ b/src/pystencils/backends/arm_instruction_sets.py @@ -1,3 +1,6 @@ +from pystencils.typing import CFunction + + def get_argument_string(function_shortcut, first=''): args = function_shortcut[function_shortcut.index('[') + 1: -1] arg_string = "(" @@ -87,7 +90,6 @@ def get_vector_instruction_set_arm(data_type='double', instruction_set='neon'): result[intrinsic_id] = prefix + name + suffix + undef + arg_string if instruction_set in ['sve', 'sme']: - from pystencils.backends.cbackend import CFunction result['width'] = CFunction(width, "int") result['intwidth'] = CFunction(intwidth, "int") else: diff --git a/src/pystencils/backends/riscv_instruction_sets.py b/src/pystencils/backends/riscv_instruction_sets.py index e456c2b8c05facda6f422d5061832331979c889f..885869a75a39af226fff71ca8b4a0effcf4e8f0c 100644 --- a/src/pystencils/backends/riscv_instruction_sets.py +++ b/src/pystencils/backends/riscv_instruction_sets.py @@ -1,3 +1,6 @@ +from pystencils.typing import CFunction + + def get_argument_string(function_shortcut, last=''): args = function_shortcut[function_shortcut.index('[') + 1: -1] arg_string = "(" @@ -78,7 +81,6 @@ def get_vector_instruction_set_riscv(data_type='double', instruction_set='rvv'): result[intrinsic_id] = prefix + name + suffix2 + arg_string - from pystencils.backends.cbackend import CFunction result['width'] = CFunction(width, "int") result['intwidth'] = CFunction(intwidth, "int")