Mario Lezcano Casado comments

Results 239 comments of


                                            Mario Lezcano Casado

[dynamo] Add support for Tensor.numpy(), torch.from_numpy() and np.meshgrid()

Note that that notebook shows how `np.mean(X, 1)` fails on this PR as it seems to try to trace through NumPy rather than PyTorch. I think this PR should at...

Stronger checks on ScanOp and ReduceOp

Addressed the comments and rebased. Will add tests tomorrow. cc @ThomasRaoux @joker-eph in case you have more comments

Stronger checks on ScanOp and ReduceOp

Added a few tests.

Stronger checks on ScanOp and ReduceOp

Fixed. There was a bit of a mix-up with RankedTensorTypes and ElementTypes. Now it should be fine.

[torchbench] Update skipped models.

@pytorchbot merge

Fix FakeTensorUpdater logic for updating fake tensors

the failure is real

Fix FakeTensorUpdater logic for updating fake tensors

@pytorchbot merge

Removing graph breaks in transforms

Note that what's going to be great for torchvision is that I expect pretty much any combination of transformation to be fused into one kernel. There is where the main...

derived dim

Regarding the constraint solver, could you guys add me if there is any call discussing that? Thank you!

[AOTDispatch] Return mutated inputs directly when keeping mutations

ping @bdhirsh