[tvm] branch main updated: Fix compilation when Arm FP16 extensions are enabled (#7386)

[mb...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

mbaret pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/main by this push:
     new 33f30af  Fix compilation when Arm FP16 extensions are enabled (#7386)
33f30af is described below

commit 33f30aff0dac784836dc4bf4e12bf1cf845a6370
Author: Matthew Bentham <ma...@arm.com>
AuthorDate: Sun Feb 7 21:47:44 2021 +0000

    Fix compilation when Arm FP16 extensions are enabled (#7386)
    
    Fixes incorrect number of template parameters in call to sort()
    
    Signed-off-by: Matthew Bentham <ma...@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 fba57d9..66f36ff 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);