From 7768a0737386d5d48bb93de7d329df13c30c6da3 Mon Sep 17 00:00:00 2001
From: Scott Wolchok <swolchok@meta.com>
Date: Mon, 2 Jun 2025 12:38:11 -0700
Subject: [PATCH] Update

[ghstack-poisoned]
---
 kernels/test/op_glu_test.cpp | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/kernels/test/op_glu_test.cpp b/kernels/test/op_glu_test.cpp
index f8bf22dae63..b18117eaa4e 100644
--- a/kernels/test/op_glu_test.cpp
+++ b/kernels/test/op_glu_test.cpp
@@ -51,20 +51,20 @@ class OpGluOutTest : public OperatorTest {
     const std::vector<int32_t> out_sizes_1 = {2, 2};
 
     // Valid input should give the expected output
-    Tensor in = tf.ones(sizes);
+    Tensor in = tf.make(sizes, {0, 1, 2, 3, 4, 5, 6, 7});
     Tensor out = tf_out.zeros(out_sizes_1);
     op_glu_out(in, 0, out);
-    expect_tensor_close<DTYPE>(
+    expect_tensor_close<OUT_DTYPE>(
         out,
         tf_out.make(
-            out_sizes_1, /*data=*/{0.731059, 0.731059, 0.731059, 0.731059}));
+            out_sizes_1, /*data=*/{0, 0.99330717, 1.99505484, 2.99726701}));
     const std::vector<int32_t> out_sizes_2 = {4, 1};
     out = tf_out.zeros(out_sizes_2);
     op_glu_out(in, 1, out);
-    expect_tensor_close<DTYPE>(
+    expect_tensor_close<OUT_DTYPE>(
         out,
         tf_out.make(
-            out_sizes_2, /*data=*/{0.731059, 0.731059, 0.731059, 0.731059}));
+            out_sizes_2, /*data=*/{0, 1.90514827, 3.97322869, 5.99453402}));
   }
 
   // Mismatched shape tests.