Skip to content

Port kernels to be structured [tracker] #55070

@ezyang

Description

@ezyang

WARNING: BEFORE STARTING ON A STRUCTURED KERNEL PORT, CHECK OPEN PULL REQUESTS TO SEE IF SOMEONE IS WORKING ON IT ALREADY. https://github.com/pytorch/pytorch/pulls?q=is%3Apr+is%3Aopen+structured+in%3Atitle

RFC at: https://github.com/pytorch/rfcs/blob/rfc-0005/RFC-0005-structured-kernel-definitions.md

Here is the list of all functions which are immediately eligible to be ported to structured (as in, they have both a functional and out version with exactly the same signatures):

Things that require adding an out kernel and then can be made structured:

  • _embedding_bag (high priority)

Issues marked (XLA) are supported by XLA and thus are higher priority to port to structured.

Functions with 1d/2d/3d in their name tend to be easier and more beginner friendly

Some of these functions may not be immediately portable. Common grounds for disqualification:

Metadata

Metadata

Assignees

No one assigned

    Labels

    module: bootcampWe plan to do a full writeup on the issue, and then get someone to do it for onboardingmodule: structured kernelsRelated to new structured kernels functionalitytrackerA tracking issuetriagedThis 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