diff --git a/src/core/graph/context.rs b/src/core/graph/context.rs index cb1a094..780ebfa 100644 --- a/src/core/graph/context.rs +++ b/src/core/graph/context.rs @@ -1,5 +1,3 @@ -use std::collections::HashMap; - use self::operation::{ ConstantBinding, Node, NodeIdentifier, Operation, Parameter, ParameterBinding, }; diff --git a/src/core/graph/mod.rs b/src/core/graph/mod.rs index dab3028..69926f4 100644 --- a/src/core/graph/mod.rs +++ b/src/core/graph/mod.rs @@ -39,7 +39,7 @@ pub fn example() { let maybe_client = xla::PjRtClient::gpu(0.7, false); let client = match maybe_client { Ok(c) => c, - Err(_) => panic!("Failed to construct XLA client!") + Err(e) => panic!("Failed to construct XLA client!\n{}", e) }; let name = "test"; let executable = ctx.compile(sum, &name, &client);