Home
last modified time | relevance | path

Searched refs:generated_kernel_count (Results 1 – 18 of 18) sorted by relevance

/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_foreach.py150 self.assertEqual(torch._inductor.metrics.generated_kernel_count, 1)
156 self.assertEqual(torch._inductor.metrics.generated_kernel_count, 1)
162 self.assertEqual(torch._inductor.metrics.generated_kernel_count, 1)
184 self.assertEqual(torch._inductor.metrics.generated_kernel_count, 1)
201 self.assertEqual(torch._inductor.metrics.generated_kernel_count, 1)
220 self.assertEqual(torch._inductor.metrics.generated_kernel_count, 1)
246 self.assertEqual(torch._inductor.metrics.generated_kernel_count, 1)
267 self.assertEqual(torch._inductor.metrics.generated_kernel_count, 1)
289 self.assertEqual(torch._inductor.metrics.generated_kernel_count, 2)
309 self.assertEqual(torch._inductor.metrics.generated_kernel_count, 2)
[all …]
H A Dtest_combo_kernels.py64 self.assertEqual(torch._inductor.metrics.generated_kernel_count, 1)
87 self.assertTrue(torch._inductor.metrics.generated_kernel_count <= 2)
110 self.assertEqual(torch._inductor.metrics.generated_kernel_count, 1)
148 self.assertEqual(torch._inductor.metrics.generated_kernel_count, 2)
185 self.assertEqual(torch._inductor.metrics.generated_kernel_count, 5)
208 self.assertTrue(4 < torch._inductor.metrics.generated_kernel_count <= 10)
231 self.assertTrue(torch._inductor.metrics.generated_kernel_count in [6, 9])
255 self.assertEqual(torch._inductor.metrics.generated_kernel_count, 6)
277 self.assertTrue(7 <= torch._inductor.metrics.generated_kernel_count <= 8)
294 self.assertEqual(torch._inductor.metrics.generated_kernel_count, 4)
[all …]
H A Dtest_loop_ordering.py224 self.assertEqual(1, metrics.generated_kernel_count)
248 self.assertEqual(1, metrics.generated_kernel_count)
258 self.assertEqual(1, metrics.generated_kernel_count)
268 self.assertEqual(1, metrics.generated_kernel_count)
284 self.assertEqual(1, metrics.generated_kernel_count)
298 self.assertEqual(2, metrics.generated_kernel_count)
353 self.assertEqual(2, metrics.generated_kernel_count)
372 self.assertEqual(1, metrics.generated_kernel_count)
394 self.assertEqual(1, metrics.generated_kernel_count)
H A Dtest_mkldnn_pattern_matcher.py325 generated_kernel_count = cal_conv_generated_kernel_number(mod, v, dtype)
326 self.assertEqual(metrics.generated_kernel_count, generated_kernel_count)
386 self.assertEqual(metrics.generated_kernel_count, 1)
441 self.assertEqual(metrics.generated_kernel_count, 1)
494 generated_kernel_count = cal_conv_generated_kernel_number(mod, v, dtype)
495 self.assertEqual(metrics.generated_kernel_count, generated_kernel_count)
565 generated_kernel_count = cal_conv_generated_kernel_number(mod, v, dtype)
566 self.assertEqual(metrics.generated_kernel_count, generated_kernel_count)
626 self.assertEqual(metrics.generated_kernel_count, 1)
H A Dtest_codecache.py407 self.assertEqual(metrics.generated_kernel_count, 0)
412 self.assertEqual(metrics.generated_kernel_count, 1)
418 self.assertEqual(metrics.generated_kernel_count, 2)
H A Dtest_compiled_optimizers.py431 torch._inductor.metrics.generated_kernel_count, kernel_count
486 torch._inductor.metrics.generated_kernel_count,
H A Dtest_torchinductor.py704 self.assertEqual(torch._inductor.metrics.generated_kernel_count, expected)
3789 torch._inductor.metrics.generated_kernel_count = 0
3824 torch._inductor.metrics.generated_kernel_count = 0
3846 torch._inductor.metrics.generated_kernel_count = 0
4127 torch._inductor.metrics.generated_kernel_count = 0
4153 torch._inductor.metrics.generated_kernel_count = 0
4233 torch._inductor.metrics.generated_kernel_count = 0
4722 print("-->", torch._inductor.metrics.generated_kernel_count)
4724 self.assertTrue(torch._inductor.metrics.generated_kernel_count > 1)
6633 torch._inductor.metrics.generated_kernel_count = 0
[all …]
H A Dtest_cpu_repro.py2085 metrics.generated_kernel_count
2755 self.assertEqual(metrics.generated_kernel_count, 3)
2767 self.assertEqual(metrics.generated_kernel_count, 3)
2779 print(metrics.generated_kernel_count)
2780 self.assertEqual(metrics.generated_kernel_count, 2)
2792 self.assertEqual(metrics.generated_kernel_count, 1)
3157 self.assertEqual(metrics.generated_kernel_count, 1)
3278 assert metrics.generated_kernel_count == 0
H A Dtest_unbacked_symints.py224 self.assertEqual(torch._inductor.metrics.generated_kernel_count, 2)
H A Dtest_torchinductor_dynamic_shapes.py944 self.assertEqual(metrics.generated_kernel_count, 0)
949 self.assertEqual(metrics.generated_kernel_count, n)
H A Dtest_triton_kernels.py397 self.assertEqual(metrics.generated_kernel_count, 1)
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dmetrics.py22 generated_kernel_count = 0 variable
57 global generated_kernel_count
67 generated_kernel_count = 0
87 generated_kernel_count: int
H A Dscheduler.py3721 old_generated_kernel_count = metrics.generated_kernel_count
3723 metrics.generated_kernel_count = old_generated_kernel_count
/aosp_15_r20/external/pytorch/benchmarks/dynamo/microbenchmarks/
H A Dbench_mm_fusion.py85 torch._inductor.metrics.generated_kernel_count == 1
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_modules.py2952 torch._inductor.metrics.generated_kernel_count = 0
2954 self.assertGreater(torch._inductor.metrics.generated_kernel_count, 0)
2966 torch._inductor.metrics.generated_kernel_count = 0
2968 opt_success = torch._inductor.metrics.generated_kernel_count == 0
2970 torch._inductor.metrics.generated_kernel_count = 0
2972 loaded_success = torch._inductor.metrics.generated_kernel_count == 0
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/
H A Dtriton_combo_kernel.py459 metrics.generated_kernel_count -= 1
H A Dcommon.py1610 metrics.generated_kernel_count += 1
H A Dcpp.py3667 metrics.generated_kernel_count -= 1