Update linear_combination_generic.h
This commit is contained in:
parent
96dad61a75
commit
e49f690fd7
@ -103,7 +103,7 @@ public:
|
|||||||
CUTLASS_HOST_DEVICE
|
CUTLASS_HOST_DEVICE
|
||||||
Params(
|
Params(
|
||||||
ElementCompute alpha,
|
ElementCompute alpha,
|
||||||
ElementCompute beta
|
ElementCompute beta = ElementCompute(0)
|
||||||
): alpha(alpha), beta(beta), alpha_ptr(nullptr), beta_ptr(nullptr) {
|
): alpha(alpha), beta(beta), alpha_ptr(nullptr), beta_ptr(nullptr) {
|
||||||
|
|
||||||
}
|
}
|
||||||
@ -111,7 +111,7 @@ public:
|
|||||||
CUTLASS_HOST_DEVICE
|
CUTLASS_HOST_DEVICE
|
||||||
Params(
|
Params(
|
||||||
ElementCompute const *alpha_ptr,
|
ElementCompute const *alpha_ptr,
|
||||||
ElementCompute const *beta_ptr
|
ElementCompute const *beta_ptr = nullptr
|
||||||
): alpha(0), beta(0), alpha_ptr(alpha_ptr), beta_ptr(beta_ptr) {
|
): alpha(0), beta(0), alpha_ptr(alpha_ptr), beta_ptr(beta_ptr) {
|
||||||
|
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user