From f83c21e5c81e790df6254402b79f7ecb6a7af8d4 Mon Sep 17 00:00:00 2001
From: Matthew Bentham <matthew.bentham@arm.com>
Date: Sun, 7 Feb 2021 21:47:44 +0000
Subject: [PATCH] Fix compilation when Arm FP16 extensions are enabled (#7386)

Fixes incorrect number of template parameters in call to sort()

Signed-off-by: Matthew Bentham <matthew.bentham@arm.com>
---
 src/runtime/contrib/sort/sort.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/runtime/contrib/sort/sort.cc b/src/runtime/contrib/sort/sort.cc
index fba57d923b38..66f36ffa50d6 100644
--- a/src/runtime/contrib/sort/sort.cc
+++ b/src/runtime/contrib/sort/sort.cc
@@ -289,7 +289,7 @@ TVM_REGISTER_GLOBAL("tvm.contrib.sort.sort").set_body([](TVMArgs args, TVMRetVal
     sort<double>(input, output, axis, is_ascend);
 #if (__ARM_FEATURE_FP16_SCALAR_ARITHMETIC == 1)
   } else if (data_dtype == "float16") {
-    sort<__fp16, __fp16>(input, output, axis, is_ascend);
+    sort<__fp16>(input, output, axis, is_ascend);
 #endif
   } else if (data_dtype == "int32") {
     sort<int32_t>(input, output, axis, is_ascend);