KEMBAR78
`torch.meshgrid` requires an `OpInfo` test · Issue #62719 · pytorch/pytorch · GitHub
Skip to content

torch.meshgrid requires an OpInfo test #62719

@dagitses

Description

@dagitses

🚀 Feature

torch.meshgrid gets an OpInfo test.

Motivation

OpInfo tests are clearly important, especially for operators that have a NumPy/SciPy reference implementation. In particular, we want to enforce compatibility with the NumPy implementation (see #50276).

Pitch

Add an OpInfo test that, among other things, enforces NumPy compatibility.

Alternatives

None: this test is important.

Additional context

This operation is also part of the Python Array API proposal: https://data-apis.org/array-api/latest/API_specification/creation_functions.html#meshgrid-arrays-indexing-xy.

cc @mruberry @rgommers @heitorschueroff @VitalyFedyunin @walterddr

Metadata

Metadata

Assignees

Labels

module: numpyRelated to numpy support, and also numpy compatibility of our operatorsmodule: testsIssues related to tests (not the torch.testing module)triagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate module

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions