Searched refs:assertGeneratedKernelCountEqual (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_torchinductor.py | 696 def assertGeneratedKernelCountEqual(self: TestCase, expected: int): function
1227 assertGeneratedKernelCountEqual(self, 1)
1264 assertGeneratedKernelCountEqual(self, 1 if self.device == GPU_TYPE else 2)
3795 assertGeneratedKernelCountEqual(self, 0)
3830 assertGeneratedKernelCountEqual(self, 0)
3852 assertGeneratedKernelCountEqual(self, 0)
4132 assertGeneratedKernelCountEqual(self, 0)
4158 assertGeneratedKernelCountEqual(self, 0)
4238 assertGeneratedKernelCountEqual(self, 0)
4555 assertGeneratedKernelCountEqual(self, 1)
[all …]
|