CUTLASS 3.3.0 (#1167)
* Release 3.3.0 Adds support for mixed precision GEMMs On Hopper and Ampere Adds support for < 16B aligned GEMMs on Hopper Enhancements to EVT Enhancements to Python interface Enhancements to Sub-byte type handling in CuTe Several other bug-fixes and performance improvements. * minor doc update
This commit is contained in:
@@ -60,4 +60,20 @@ struct TrivialPredTensor
|
||||
}
|
||||
};
|
||||
|
||||
template <class Fn>
|
||||
struct FunctionPredTensor
|
||||
{
|
||||
CUTE_HOST_DEVICE constexpr
|
||||
FunctionPredTensor(Fn const& fn) : fn_(fn) {}
|
||||
|
||||
template <class... Coords>
|
||||
CUTE_HOST_DEVICE constexpr
|
||||
auto
|
||||
operator()(Coords const&... coords) const {
|
||||
return fn_(coords...);
|
||||
}
|
||||
|
||||
Fn const& fn_;
|
||||
};
|
||||
|
||||
} // end namespace cute
|
||||
|
||||
Reference in New Issue
Block a user