Skip to content

Unify all pallas kernel code in one directory  #8676

Open
@miladm

Description

🐛 Bug

Please move https://github.com/pytorch/xla/blob/master/torch_xla/experimental/custom_kernel.py to the https://github.com/pytorch/xla/blob/master/torch_xla/experimental/pallas_kernels directory to unify all kernel code in one location.

Let's do the same for the test codes under /test director.

Metadata

Assignees

Labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions