Class KernelRegistryStatistics

tools/parse_kernel_info.py:56–88

class KernelRegistryStatistics:
    def __init__(self):
        self.num_ops_for_dtypes = {
            "all": 0,
            "float32": 0,
            "float16": 0,
            "bfloat16": 0,
        }

    def update(self, 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):
        res = "{ "
        num_floats = int(self.num_ops_for_dtypes["float32"])
        for dtype, num in self.num_ops_for_dtypes.items():
            res += f"{dtype}: {num:4d}"
            if dtype in ["float16", "bfloat16"]:
                if num_floats != 0:
                    percent = float(self.num_ops_for_dtypes[dtype]) / float(
                        num_floats
                    )
                    res += f"({percent * 100:.2f}%)"
                else:
                    res += f"({0:.2f}%)"
            res += " "
        res += "}"
        return res


def parse_paddle_kernels(lib="phi", kernel_type="function", print_detail=False):