Skip to content

Commit e272cfd

Browse files
committed
Change python interface to return a set
Change-Id: If41e4fd32947a2acddfe9b0691a0c9ba3245d722
1 parent 5e4ba6a commit e272cfd

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

python/tvm/target/codegen.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -183,7 +183,8 @@ def llvm_get_cpu_features(target=None):
183183
List of available CPU features.
184184
"""
185185
assert isinstance(target, Target) or target is None
186-
return _ffi_api.llvm_get_cpu_features(target)
186+
feature_map = _ffi_api.llvm_get_cpu_features(target)
187+
return set(feature_map.keys())
187188

188189

189190
def llvm_cpu_has_features(cpu_features, target=None):

0 commit comments

Comments
 (0)