squall/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Add missing CUTLASS_HOST_DEVICE (#671)

Browse Source
This commit is contained in:
hlu1 2022-10-21 19:20:38 -07:00 committed by GitHub
parent 4db6a6140e
commit 9b47403b2d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
include/cutlass/epilogue/thread/activation.h
View File

@ -347,6 +347,7 @@ struct Sigmoid<Array<half_t, N>> {
using Params = LinearCombinationGenericParams<T>;
CUTLASS_HOST_DEVICE
Array<T, N> operator()(Array<T, N> const &z, Params const &params_) const {
return this->operator()(z);
}