Tensors and Dynamic neural networks in Python with strong GPU acceleration - [Dynamo] Replace `torch._dynamo.optimize()` with `torch.compile()` [8/N] · pytorch/pytorch@e86bbbd
Tensors and Dynamic neural networks in Python with strong GPU acceleration - [Dynamo] Replace `torch._dynamo.optimize()` with `torch.compile()` [1/N] · pytorch/pytorch@a2ac96c
x = torch.randn(4) aot_fn = torch._dynamo.optimize("aot_eager")(fn) aot_fn = torch.compile(fn, backend="aot_eager") aot_fn(x, y) def test_call_fn_with_non_const_inputs_aot_safe(self): @@ -173,7 +173,7 @@ def forward(self, x): # Run exported graph with AOT self.as...
🐛 Describe the bug I am trying to optimize a code that calls the radius function from pytorch_cluster: import torch from torch_cluster import radius import torch._dynamo as dynamo def myradius(x: torch.Tensor, y: torch.Tensor, r: float, ...
Tensors and Dynamic neural networks in Python with strong GPU acceleration - [Dynamo] Replace `torch._dynamo.optimize()` with `torch.compile()` [1/N] · pytorch/pytorch@a2ac96c
Tensors and Dynamic neural networks in Python with strong GPU acceleration - [Dynamo] Replace `torch._dynamo.optimize()` with `torch.compile()` [4/N] · pytorch/pytorch@565a794
Tensors and Dynamic neural networks in Python with strong GPU acceleration - [Dynamo] Replace `torch._dynamo.optimize()` with `torch.compile()` [7/N] · pytorch/pytorch@6b0df2f
Tensors and Dynamic neural networks in Python with strong GPU acceleration - [Dynamo] Replace `torch._dynamo.optimize()` with `torch.compile()` [7/N] · pytorch/pytorch@e9d4a93
Tensors and Dynamic neural networks in Python with strong GPU acceleration - [Dynamo] Replace `torch._dynamo.optimize()` with `torch.compile()` [7/N] · pytorch/pytorch@e9d4a93
Tensors and Dynamic neural networks in Python with strong GPU acceleration - [Dynamo] Replace `torch._dynamo.optimize()` with `torch.compile()` [4/N] · pytorch/pytorch@d3685bc