docs: add the new xla_device (#968)

LuxDL · Oct 3, 2024 · 5d51ad8 · 5d51ad8
1 parent b7931d1
commit 5d51ad8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/docs/Project.toml b/docs/Project.toml
@@ -45,7 +45,7 @@ LuxCUDA = "0.3.2"
 LuxCore = "1"
 LuxLib = "1"
 LuxTestUtils = "1.2"
-MLDataDevices = "1.1"
+MLDataDevices = "1.2"
 Optimisers = "0.3.3"
 Pkg = "1.10"
 Printf = "1.10"

diff --git a/docs/src/api/Accelerator_Support/MLDataDevices.md b/docs/src/api/Accelerator_Support/MLDataDevices.md
@@ -18,6 +18,7 @@ MLDataDevices.gpu_backend!
 ```@docs
 MLDataDevices.cpu_device
 MLDataDevices.gpu_device
+MLDataDevices.xla_device
 ```
 
 ## Miscellaneous