pytorch/torch/csrc/mps
Yu, Guangye b0810168a3 Generalize poison fork logic for each device backend (#144664)
# Motivation
Generalize the posion_fork code to make it reusable across different devices.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/144664
Approved by: https://github.com/EikanWang, https://github.com/albanD
2025-04-13 09:54:30 +00:00
..
Module.cpp Generalize poison fork logic for each device backend (#144664) 2025-04-13 09:54:30 +00:00
Module.h [MPS] Add CompileShader method (#141478) 2024-12-11 02:00:51 +00:00