Fix compilation when Arm FP16 extensions are enabled (apache#7386)

Fixes incorrect number of template parameters in call to sort() Signed-off-by: Matthew Bentham <matthew.bentham@arm.com>
alexwong · Feb 11, 2021 · f83c21e · f83c21e
1 parent 268d262
commit f83c21e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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);