Skip to content
Snippets Groups Projects

Various extensions to the vectorizer

Merged Daniel Bauer requested to merge hyteg/pystencils:bauerd/vec-extensions into v2.0-dev
Viewing commit 221787c1
Show latest version
1 file
+ 12
7
Preferences
Compare changes
@@ -193,10 +193,15 @@ class X86VectorCpu(GenericVectorCpu):
| MathFunctions.ATan
| MathFunctions.ATan2
| MathFunctions.Pow
| MathFunctions.Floor
| MathFunctions.Ceil
) if vtype.is_float():
):
raise MaterializationError(
"Trigonometry, exp, log, and pow require SVML."
)
case MathFunctions.Floor | MathFunctions.Ceil if vtype.is_float():
opstr = expr.function.func.function_name
if vtype.width > 256:
raise MaterializationError("512bit ceil/floor require SVML.")
case MathFunctions.Min | MathFunctions.Max:
opstr = expr.function.func.function_name
@@ -234,10 +239,10 @@ class X86VectorCpu(GenericVectorCpu):
return func(neg_zero, op)
case _:
raise MaterializationError(
f"x86/{self} does not support abs on type {vtype}."
)
case _:
raise MaterializationError(
f"x86/{self} does not support {expr.function.func.function_name} on type {vtype}."
)
if expr.function.func in [
MathFunctions.ATan2,