From 6335242def4b91899b7e8c742f9215f207d910b9 Mon Sep 17 00:00:00 2001
From: Markus Holzer <markus.holzer@fau.de>
Date: Mon, 15 Jul 2024 09:02:28 +0200
Subject: [PATCH] Small fixes
---
src/pystencils/backends/cbackend.py | 4 +---
tests/test_vectorization_specific.py | 4 ++--
2 files changed, 3 insertions(+), 5 deletions(-)
diff --git a/src/pystencils/backends/cbackend.py b/src/pystencils/backends/cbackend.py
index e95fb579..10ef12e8 100644
--- a/src/pystencils/backends/cbackend.py
+++ b/src/pystencils/backends/cbackend.py
@@ -614,9 +614,7 @@ class VectorizedCustomSympyPrinter(CustomSympyPrinter):
return None
def _print_Abs(self, expr):
- if 'abs' in self.instruction_set:
- return self.instruction_set['abs'].format(self._print(expr.args[0]), **self._kwargs)
- return super()._print_Abs(expr)
+ return self.instruction_set['abs'].format(self._print(expr.args[0]), **self._kwargs)
def _typed_vectorized_number(self, expr, data_type):
basic_data_type = data_type.base_type
diff --git a/tests/test_vectorization_specific.py b/tests/test_vectorization_specific.py
index 05a62476..e8b34a2a 100644
--- a/tests/test_vectorization_specific.py
+++ b/tests/test_vectorization_specific.py
@@ -39,7 +39,7 @@ def test_vectorisation_varying_arch(instruction_set):
@pytest.mark.parametrize('dtype', ('float32', 'float64'))
@pytest.mark.parametrize('instruction_set', supported_instruction_sets)
-def test_vectorized_abs(instruction_set, dtype):
+def test_vectorized_abs_field(instruction_set, dtype):
"""Some instructions sets have abs, some don't.
Furthermore, the special treatment of unary minus makes this data type-sensitive too.
"""
@@ -59,7 +59,7 @@ def test_vectorized_abs(instruction_set, dtype):
@pytest.mark.parametrize('instruction_set', supported_instruction_sets)
-def test_vectorized_abs(instruction_set):
+def test_vectorized_abs_scalar(instruction_set):
"""Some instructions sets have abs, some don't.
Furthermore, the special treatment of unary minus makes this data type-sensitive too.
"""
--
GitLab