From cfb60fa584b7bd409a837c899b95c7958488e5ac Mon Sep 17 00:00:00 2001
From: Will Cromar <wcromar@google.com>
Date: Thu, 18 Jan 2024 22:57:38 +0000
Subject: [PATCH] GpuPlugin -> CudaPlugin

---
 plugins/cuda/README.md      | 2 +-
 plugins/cuda/pyproject.toml | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/plugins/cuda/README.md b/plugins/cuda/README.md
index a8da474948e..89d14c737ea 100644
--- a/plugins/cuda/README.md
+++ b/plugins/cuda/README.md
@@ -29,7 +29,7 @@ import torch_xla.runtime as xr
 
 # Use dynamic plugin instead of built-in CUDA support
 plugins.use_dynamic_plugins()
-plugins.register_plugin('CUDA', torch_xla_cuda_plugin.GpuPlugin())
+plugins.register_plugin('CUDA', torch_xla_cuda_plugin.CudaPlugin())
 xr.set_device_type('CUDA')
 
 print(xm.xla_device())
diff --git a/plugins/cuda/pyproject.toml b/plugins/cuda/pyproject.toml
index 3ae63c01142..fd8bbf59f6c 100644
--- a/plugins/cuda/pyproject.toml
+++ b/plugins/cuda/pyproject.toml
@@ -15,4 +15,4 @@ requires-python = ">=3.8"
 torch_xla_cuda_plugin = ["lib/*.so"]
 
 [project.entry-points."torch_xla.plugins"]
-gpu = "torch_xla_cuda_plugin:GpuPlugin"
+cuda = "torch_xla_cuda_plugin:CudaPlugin"