squash with autumnai#37

alexandermorozov · Apr 17, 2016 · 728603e · 728603e
1 parent 6c12488
commit 728603e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/tensor.rs b/src/tensor.rs
@@ -458,7 +458,7 @@ impl<T> SharedTensor<T> {
     }
 
     /// Drops memory allocation on specified device. Returns error
-    pub fn drop_at(&mut self, device: &DeviceType) -> Result<(), Error> {
+    pub fn drop_device(&mut self, device: &DeviceType) -> Result<(), Error> {
         match self.get_location_index(device) {
             Some(i) => {
                 self.locations.borrow_mut().remove(i);
@@ -492,7 +492,7 @@ impl<T> SharedTensor<T> {
 }
 
 /// Errors than can occur when synchronizing memory.
-#[derive(Debug, Copy, Clone)]
+#[derive(Debug, Copy, Clone, PartialEq, Eq)]
 pub enum Error {
     /// No copy on source device.
     MissingSource(&'static str),