CUTLASS 3.5.1 (#1623)

* CUTLASS 3.5.1

* updates, optimizations, fixes
This commit is contained in:
Vijay Thakkar
2024-07-29 08:46:24 -04:00
committed by GitHub
parent 56b46e2d13
commit be60a0b272
312 changed files with 19793 additions and 6775 deletions

View File

@@ -785,6 +785,24 @@ struct reciprocal_approximate<Array<T, N>> {
}
};
template <typename T, int N>
struct reciprocal_approximate_ftz<Array<T, N>> {
CUTLASS_HOST_DEVICE
Array<T, N> operator()(Array<T, N> const &lhs) const {
Array<T, N> result;
reciprocal_approximate_ftz<T> scalar_op;
CUTLASS_PRAGMA_UNROLL
for (int i = 0; i < N; ++i) {
result[i] = scalar_op(lhs[i]);
}
return result;
}
};
template <typename T, int N>
struct maximum<Array<T, N>, false> {
@@ -979,6 +997,10 @@ struct minimum<Array<T, N>, true> {
}
};
template <typename T, int N>
struct minimum_with_nan_propagation<Array<T, N>> : minimum<Array<T, N>, true>
{};
template <typename T, int N>
struct negate<Array<T, N>> {