diff --git a/tools/parse_kernel_info.py b/tools/parse_kernel_info.py index b3b4203daef6902e6371ef3f1cd60dc1c6f5ba1f..ef85a0bb4bbd6fa011a766c7e928fc3de0ba69ec 100644 --- a/tools/parse_kernel_info.py +++ b/tools/parse_kernel_info.py @@ -63,11 +63,11 @@ class KernelRegistryStatistics: } def update(self, supported_dtypes): - for dtype in self.num_ops_for_dtypes.keys(): - if dtype in ["float", "float32"]: - self.num_ops_for_dtypes["float32"] += 1 - elif dtype in supported_dtypes: + for dtype in supported_dtypes: + if dtype in self.num_ops_for_dtypes.keys(): self.num_ops_for_dtypes[dtype] += 1 + elif dtype == "float": + self.num_ops_for_dtypes["float32"] += 1 self.num_ops_for_dtypes["all"] += 1 def __str__(self):