From ac1600eb32c8fda16d71b7545282219ca40bb790 Mon Sep 17 00:00:00 2001
From: huijjj <huijong.jeong@squeezebits.com>
Date: Wed, 24 Apr 2024 07:17:46 +0000
Subject: [PATCH] fix: input description must have unique index

---
 scripts/pytorch_custom_op/hpu_custom_searchsorted.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/scripts/pytorch_custom_op/hpu_custom_searchsorted.cpp b/scripts/pytorch_custom_op/hpu_custom_searchsorted.cpp
index 2f74947..2f274a4 100644
--- a/scripts/pytorch_custom_op/hpu_custom_searchsorted.cpp
+++ b/scripts/pytorch_custom_op/hpu_custom_searchsorted.cpp
@@ -17,9 +17,9 @@ bool register_custom_searchsorted() {
     habana::custom_op::InputDesc input_a_desc{
         habana::custom_op::input_type::TENSOR, 0};
     habana::custom_op::InputDesc input_b_desc{
-        habana::custom_op::input_type::TENSOR, 0};
+        habana::custom_op::input_type::TENSOR, 1};
     habana::custom_op::InputDesc input_c_desc{
-        habana::custom_op::input_type::SCALAR, 0};
+        habana::custom_op::input_type::SCALAR, 2};
 
     std::vector<habana::custom_op::InputDesc> inputs_desc{
         input_a_desc, input_b_desc, input_c_desc};