def ck_lib_layouts():
        """
        return [layout_a, layout_b, layout_c] in the form of ck_lib definitions
        """
        import ck_lib

        return [
            ck_lib.library.LayoutType.RowMajor,
            ck_lib.library.LayoutType.RowMajor,
            ck_lib.library.LayoutType.RowMajor,
        ]

        return [layout_a, layout_b, layout_c] in the form of ck_lib definitions
        
python/aitemplate/backend/rocm/gemm/layout.py