tweak

8fecf4c3 · Chao Liu · 6ff00ed4 · 8fecf4c3
Commit 8fecf4c3 authored Jan 17, 2022 by Chao Liu
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

example/3_gemm_xdl_bias_relu_add/gemm_xdl_bias_relu_add.cpp example/3_gemm_xdl_bias_relu_add/gemm_xdl_bias_relu_add.cpp +2 -2

No files found.
--- a/example/3_gemm_xdl_bias_relu_add/gemm_xdl_bias_relu_add.cpp
+++ b/example/3_gemm_xdl_bias_relu_add/gemm_xdl_bias_relu_add.cpp
@@ -35,9 +35,9 @@ struct BiasReluAdd
    template <typename T1, typename T2>
    __device__ constexpr float operator()(float v0, T1 v1, T2 v2) const
    {
-#if 0
+#if 1
        float a = v1 + v0;
-        float b = max(a, float(0));
+        float b = a > 0 ? a : 0;
        float c = b + v2;
        return c;