Fix Vectorized<double>::next_after SVE compilation (#136388)

malfet · pytorchmergebot · commit 7936584a8830 · 2024-09-20T23:54:17.000Z
Should have called [`Sleef_nextafterdx_sve`](https://sleef.org/2-references/libm/aarch64#vectorized-double-precision-function-for-obtaining-the-next-representable-fp-value) rather than [`Sleef_nextafterfx_sve`](https://sleef.org/2-references/libm/aarch64#vectorized-single-precision-function-for-obtaining-the-next-representable-fp-value) to get vectorized `nextafter` for double precision rather than single precision values This fixes a compilation issue introduced by #119571 and exposed by #133339 Pull Request resolved: #136388 Approved by: https://github.com/kit1980
diff --git a/aten/src/ATen/cpu/vec/sve/vec_double.h b/aten/src/ATen/cpu/vec/sve/vec_double.h
@@ -261,7 +261,7 @@ template <> class Vectorized<double> {
   Vectorized<double> nextafter(const Vectorized<double> &b) const {
     USE_SLEEF(
       {
-        return Vectorized<double>(Sleef_nextafterfx_sve(values, b));
+        return Vectorized<double>(Sleef_nextafterdx_sve(values, b));
       },
       {
         __at_align__ double tmp[size()];

Original file line number	Diff line number	Diff line change
`@@ -261,7 +261,7 @@ template <> class Vectorized<double> {`
`261`	`261`	`Vectorized<double> nextafter(const Vectorized<double> &b) const {`
`262`	`262`	`USE_SLEEF(`
`263`	`263`	`{`
`264`		`- return Vectorized<double>(Sleef_nextafterfx_sve(values, b));`
	`264`	`+ return Vectorized<double>(Sleef_nextafterdx_sve(values, b));`
`265`	`265`	`},`
`266`	`266`	`{`
`267`	`267`	`__at_align__ double tmp[size()];`