Class KernelInfo

tools/parse_kernel_info.py:18–53

class KernelInfo:
    def __init__(self, op_type):
        self.op_type = op_type
        self.supported_dtypes = set()

    def parse_phi_dtypes(self, registered_info_list, device="GPU"):
        assert isinstance(registered_info_list, list)
        assert device in ["CPU", "GPU"]

        # registered_info_list is in format as follows:
        # ['(GPU, Undefined(AnyLayout), float32)', '(GPU, Undefined(AnyLayout), float64)']
        for kernel_str in registered_info_list:
            kernel_strs = (
                kernel_str.replace("(", "").replace(")", "").split(",")
            )
            # for GPU, the kernel type can be GPUDNN.
            if device in kernel_strs[0]:
                self.supported_dtypes.add(kernel_strs[-1].replace(" ", ""))

        # if len(self.supported_dtypes) == 0:
        #    print("-- [WARNING] No dtypes for op_type={}, device={}. Registered info: {}".format(self.op_type, device, registered_info_list))

    def parse_fluid_dtypes(self, registered_info_list, device="gpu"):
        assert isinstance(registered_info_list, list)
        assert device in ["cpu", "gpu"]

        # registered_info_list is in format as follows:
        # ['{data_type[::paddle::platform::bfloat16]; data_layout[Undefined(AnyLayout)]; place[Place(gpu:0)]; library_type[PLAIN]}', ...}']
        for kernel_str in registered_info_list:
            kernel_strs = kernel_str.split(";")
            if "place" in kernel_strs[2] and device in kernel_strs[2]:
                assert "data_type" in kernel_strs[0]
                dtype_str = kernel_strs[0].replace("{data_type[", "")
                dtype_str = dtype_str.replace("::paddle::platform::", "")
                dtype_str = dtype_str.replace("]", "")
                self.supported_dtypes.add(dtype_str)


class KernelRegistryStatistics: