Add missing #include directive to get access to cutlass::epilogue:🧵:ScaleType. (#925)

Currently, the `LinearCombinationClamp` header file is not standalone,
and must have the definition of `cutlass::epilogue:🧵:ScaleType`
already available when it is `#include`d.
This commit is contained in:
Gregory Meyer (gregjm) 2023-04-28 17:02:41 -07:00 committed by GitHub
parent fe2f491dd7
commit 6f8596ce3f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -40,6 +40,7 @@
#include "cutlass/array.h"
#include "cutlass/functional.h"
#include "cutlass/numeric_conversion.h"
#include "cutlass/epilogue/thread/scale_type.h"
/////////////////////////////////////////////////////////////////////////////////////////////////