Home
last modified time | relevance | path

Searched refs:complex_memory_overlap (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_cpu_repro.py31 complex_memory_overlap,
1557 self.assertFalse(complex_memory_overlap(dense))
1558 self.assertFalse(complex_memory_overlap(dense.t()))
1561 self.assertFalse(complex_memory_overlap(strided[0]))
1562 self.assertFalse(complex_memory_overlap(strided[0].t()))
1565 self.assertFalse(complex_memory_overlap(unsqueezed))
1566 self.assertFalse(complex_memory_overlap(unsqueezed.permute(1, 2, 0)))
1569 self.assertFalse(complex_memory_overlap(gathered))
1570 self.assertFalse(complex_memory_overlap(gathered.t()))
H A Dtest_cudagraph_trees.py602 prev = torch._inductor.compile_fx.complex_memory_overlap
603 torch._inductor.compile_fx.complex_memory_overlap = (
609 torch._inductor.compile_fx.complex_memory_overlap = prev
H A Dtest_torchinductor.py113 complex_memory_overlap,
10461 self.assertFalse(complex_memory_overlap(t))
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dcompile_fx.py106 def complex_memory_overlap(t: torch.Tensor) -> bool: function
570 complex_memory_overlap(t)