MCPcopy Create free account
hub / github.com/PaddlePaddle/Paddle / main

Function main

tools/parse_kernel_info.py:138–171  ·  view source on GitHub ↗
(lib)

Source from the content-addressed store, hash-verified

136
137
138def main(lib):
139 assert lib in ["fluid", "phi"]
140
141 print_detail = False
142 if lib == "phi":
143 phi_function_kernels_stats = parse_paddle_kernels(
144 lib, "function", print_detail=False
145 )
146 phi_structure_kernels_stats = parse_paddle_kernels(
147 lib, "structure", print_detail=False
148 )
149 phi_all_kernels_stats = parse_paddle_kernels(
150 lib, "all", print_detail=print_detail
151 )
152 print(
153 "================================== phi kernels summary =================================="
154 )
155 print(f"phi function kernels : {phi_function_kernels_stats}")
156 print(f"phi structure kernels : {phi_structure_kernels_stats}")
157 print(f"phi all kernels : {phi_all_kernels_stats}")
158 print()
159 else:
160 fluid_ops_stats = parse_paddle_kernels(lib, "fluid", print_detail=False)
161 phi_ops_stats = parse_paddle_kernels(lib, "phi", print_detail=False)
162 all_ops_stats = parse_paddle_kernels(
163 lib, "all", print_detail=print_detail
164 )
165 print(
166 "================================== fluid operators summary =================================="
167 )
168 print(f"fluid operators : {fluid_ops_stats}")
169 print(f"phi operators : {phi_ops_stats}")
170 print(f"all operators : {all_ops_stats}")
171 print()
172
173
174main(lib="fluid")

Callers 1

Calls 2

parse_paddle_kernelsFunction · 0.85
printFunction · 0.50

Tested by

no test coverage detected