From 5d51ad8cd4fd41333f9dd094b947f286f0aec136 Mon Sep 17 00:00:00 2001 From: Avik Pal Date: Thu, 3 Oct 2024 16:20:42 -0400 Subject: [PATCH] docs: add the new `xla_device` (#968) --- docs/Project.toml | 2 +- docs/src/api/Accelerator_Support/MLDataDevices.md | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/docs/Project.toml b/docs/Project.toml index 8ec8cc050..e941459ad 100644 --- 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 index 1009bed6c..d375c7fcb 100644 --- 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