We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
REACTANT_DEFAULT_DEVICE
1 parent 9528846 commit 386197eCopy full SHA for 386197e
src/xla/Client.jl
@@ -13,4 +13,5 @@ function get_device end
13
function get_addressable_device end
14
function platform_name end
15
16
-default_device(client::AbstractClient) = first(addressable_devices(client))
+default_device(client::AbstractClient) =
17
+ addressable_devices(client)[something(tryparse(Int, get(ENV, "REACTANT_DEFAULT_DEVICE", "1")), 1)]
0 commit comments