Change python interface to return a set

Change-Id: If41e4fd32947a2acddfe9b0691a0c9ba3245d722
apache · Jan 22, 2024 · e272cfd · e272cfd
1 parent 5e4ba6a
commit e272cfd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/tvm/target/codegen.py b/python/tvm/target/codegen.py
@@ -183,7 +183,8 @@ def llvm_get_cpu_features(target=None):
         List of available CPU features.
     """
     assert isinstance(target, Target) or target is None
-    return _ffi_api.llvm_get_cpu_features(target)
+    feature_map = _ffi_api.llvm_get_cpu_features(target)
+    return set(feature_map.keys())
 
 
 def llvm_cpu_has_features(cpu_features, target=None):