diff options
Diffstat (limited to 'kernels/src/binary_add.cu')
| -rw-r--r-- | kernels/src/binary_add.cu | 18 |
1 files changed, 3 insertions, 15 deletions
diff --git a/kernels/src/binary_add.cu b/kernels/src/binary_add.cu index 16156bc2..979a429e 100644 --- a/kernels/src/binary_add.cu +++ b/kernels/src/binary_add.cu @@ -1,20 +1,8 @@ #include "binary_op_macros.cuh" -struct BinaryAddOp {}; - #if __CUDA_ARCH__ >= 530 -BINARY_OP(__half, badd_fwd_f16, badd_bwd_lhs_f16, badd_bwd_rhs_f16, BinaryAddOp, - x + y, - 1.0, - 1.0) +BINARY_OP(__half, badd_f16, x + y) #endif -BINARY_OP(float, badd_fwd_f32, badd_bwd_lhs_f32, badd_bwd_rhs_f32, BinaryAddOp, - x + y, - 1.0, - 1.0) - -BINARY_OP(double, badd_fwd_f64, badd_bwd_lhs_f64, badd_bwd_rhs_f64, BinaryAddOp, - x + y, - 1.0, - 1.0) +BINARY_OP(float, badd_f32, x + y) +BINARY_OP(double, badd_fwd_f64, x + y); |