/aosp_15_r20/external/pytorch/test/inductor/ |
H A D | test_foreach.py | 150 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 D | test_combo_kernels.py | 64 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 D | test_loop_ordering.py | 224 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 D | test_mkldnn_pattern_matcher.py | 325 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 D | test_codecache.py | 407 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 D | test_compiled_optimizers.py | 431 torch._inductor.metrics.generated_kernel_count, kernel_count 486 torch._inductor.metrics.generated_kernel_count,
|
H A D | test_torchinductor.py | 704 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 D | test_cpu_repro.py | 2085 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 D | test_unbacked_symints.py | 224 self.assertEqual(torch._inductor.metrics.generated_kernel_count, 2)
|
H A D | test_torchinductor_dynamic_shapes.py | 944 self.assertEqual(metrics.generated_kernel_count, 0) 949 self.assertEqual(metrics.generated_kernel_count, n)
|
H A D | test_triton_kernels.py | 397 self.assertEqual(metrics.generated_kernel_count, 1)
|
/aosp_15_r20/external/pytorch/torch/_inductor/ |
H A D | metrics.py | 22 generated_kernel_count = 0 variable 57 global generated_kernel_count 67 generated_kernel_count = 0 87 generated_kernel_count: int
|
H A D | scheduler.py | 3721 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 D | bench_mm_fusion.py | 85 torch._inductor.metrics.generated_kernel_count == 1
|
/aosp_15_r20/external/pytorch/test/dynamo/ |
H A D | test_modules.py | 2952 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 D | triton_combo_kernel.py | 459 metrics.generated_kernel_count -= 1
|
H A D | common.py | 1610 metrics.generated_kernel_count += 1
|
H A D | cpp.py | 3667 metrics.generated_kernel_count -= 1
|