remove duplicate output

pytorch · Feb 6, 2023 · 2442f06 · 2442f06
1 parent e3ac450
commit 2442f06
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/userbenchmark/api-coverage/__init__.py b/userbenchmark/api-coverage/__init__.py
@@ -67,8 +67,10 @@ def check_func_in_APIs(self, func):
         module_name, func_name = parse_func(func)
         if (module_name, func_name) not in API_LIST:
             if (module_name, func_name) not in IGNORED_API_LIST and module_name != 'torch._ops.profiler':
-                print("not in API_LIST or IGNORED_API_LIST: (%s, %s)" % (module_name, func_name))
-                self.api_need_support.add((module_name, func_name))
+                new_pair = (module_name, func_name)
+                if new_pair not in self.api_need_support:
+                    print("not in API_LIST or IGNORED_API_LIST: (%s, %s)" % (module_name, func_name))
+                    self.api_need_support.add((module_name, func_name))
         else:
             self.api_used.add((module_name, func_name))
             # debug