From 711dafdf8c3d52e0327e18d86e7769897318c2b8 Mon Sep 17 00:00:00 2001
From: "Sevin F. Varoglu" <sfvaroglu@octoml.ai>
Date: Tue, 29 Mar 2022 09:53:50 -0700
Subject: [PATCH] [dyn.nn.pad] cast pad value to input dtype

---
 src/relay/op/dyn/nn/pad.cc | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/relay/op/dyn/nn/pad.cc b/src/relay/op/dyn/nn/pad.cc
index 42ec784f8c15..101ad5de7f57 100644
--- a/src/relay/op/dyn/nn/pad.cc
+++ b/src/relay/op/dyn/nn/pad.cc
@@ -77,7 +77,8 @@ Array<te::Tensor> PadCompute(const Attrs& attrs, const Array<te::Tensor>& inputs
   auto data = inputs[0];
   auto pad_width = inputs[1];
 
-  const PrimExpr& pad_value = inputs[2](Array<PrimExpr>());
+  te::Tensor cast_pad_value = topi::cast(inputs[2], inputs[0]->dtype);
+  const PrimExpr& pad_value = cast_pad_value(Array<PrimExpr>());
 
   Array<IndexExpr> pad_before;
   Array<IndexExpr> pad_after;