Home
last modified time | relevance | path

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

/external/pytorch/test/inductor/
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()))
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
Dtest_torchinductor.py113 complex_memory_overlap,
10461 self.assertFalse(complex_memory_overlap(t))
/external/pytorch/torch/_inductor/
Dcompile_fx.py106 def complex_memory_overlap(t: torch.Tensor) -> bool: function
570 complex_memory_overlap(t)