torch-rb 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 38e16e7f07d004fd9625f168694356d551c79cbc62b0131fe1403e4c0995f296
-  data.tar.gz: 66bf6ae0e4dd373a7542fbfb1cfb9dbd89fc455e16166e6a76d0945b32fecf38
+  metadata.gz: ff0920ba955063c03309fdb45ecf228b51c556508bea30b510d6bf652c1d0b18
+  data.tar.gz: 481dccf6a8e929230033f74c82bc9d292ef38ea219e2cb2cc61ca0b0c5457403
 SHA512:
-  metadata.gz: d100e3a21ac877fe93ac61e9b5e0d8a5e61126684fc037dda3e9f703b040188b1e1523aa4111dff4aaf92ada1001597c5f60674b9583b14d31afd18dbf1ff18d
-  data.tar.gz: c234dee79e26d3ee25ade2aaddd75f155dea6d59d8b9c5af2c571423a7aaa8a6489f5cfce89f09f390468a951b1644a4212c19525a79816be09214f0938860a8
+  metadata.gz: cd6c8fd9db4af15640217c09813c4f86d3f66360202d30711015c4f34552853ef281d5614fd78dc274d405da0d9f46f08a2359475ae1b0721143db49183faf5d
+  data.tar.gz: ee638c08458e0d2a8fac52e29c45d1347a74847ca7d8dab3a9a573afd887814d4c578c4e1a7fb80b204222785b27e21ca3c138f3face681e98e63c1bc02a9a7f

data/CHANGELOG.md

@@ -1,3 +1,9 @@
+## 0.1.1 (2019-11-26)
+
+- Added support for `uint8` and `int8` types
+- Fixed `undefined symbol` error on Linux
+- Fixed C++ error messages
+
 ## 0.1.0 (2019-11-26)
 
 - First release

data/README.md

@@ -2,14 +2,16 @@
 
 :fire: Deep learning for Ruby, powered by [LibTorch](https://pytorch.org)
 
-**Note:** This gem is currently experimental. There may be breaking changes between each release.
+This gem is currently experimental. There may be breaking changes between each release.
+
+[![Build Status](https://travis-ci.org/ankane/torch-rb.svg?branch=master)](https://travis-ci.org/ankane/torch-rb)
 
 ## Installation
 
 First, [install LibTorch](#libtorch-installation). For Homebrew, use:
 
 ```sh
-brew install ankane/brew/libtorch
+brew install libtorch
 ```
 
 Add this line to your application’s Gemfile:
@@ -171,7 +173,7 @@ out.backward
 Get gradients
 
 ```ruby
-x.grad
+x.grad # tensor([[4.5, 4.5], [4.5, 4.5]])
 ```
 
 Stop autograd from tracking history
@@ -242,7 +244,7 @@ Here’s a list of functions to create tensors (descriptions from the [C++ docs]
 - `empty` returns a tensor with uninitialized values
 
   ```ruby
-  Torch.empty(3)
+  Torch.empty(3) # tensor([7.0054e-45, 0.0000e+00, 0.0000e+00])
   ```
 
 - `eye` returns an identity matrix
@@ -278,19 +280,19 @@ Here’s a list of functions to create tensors (descriptions from the [C++ docs]
 - `rand` returns a tensor filled with values drawn from a uniform distribution on [0, 1)
 
   ```ruby
-  Torch.rand(3)
+  Torch.rand(3) # tensor([0.5444, 0.8799, 0.5571])
   ```
 
 - `randint` returns a tensor with integers randomly drawn from an interval
 
   ```ruby
-  Torch.randint(1, 10, [3])
+  Torch.randint(1, 10, [3]) # tensor([7, 6, 4])
   ```
 
 - `randn` returns a tensor filled with values drawn from a unit normal distribution
 
   ```ruby
-  Torch.randn(3)
+  Torch.randn(3) # tensor([-0.7147,  0.6614,  1.1453])
   ```
 
 - `randperm` returns a tensor filled with a random permutation of integers in some interval
@@ -307,10 +309,10 @@ Here’s a list of functions to create tensors (descriptions from the [C++ docs]
 
 ## LibTorch Installation
 
-[Download LibTorch](https://pytorch.org/) and run:
+[Download LibTorch](https://pytorch.org/). For Linux, use the `cxx11 ABI` version. Then run:
 
 ```sh
-gem install torch-rb -- --with-torch-dir=/path/to/libtorch
+bundle config build.torch-rb --with-torch-dir=/path/to/libtorch
 ```
 
 ### Homebrew
@@ -318,10 +320,10 @@ gem install torch-rb -- --with-torch-dir=/path/to/libtorch
 For Mac, you can use Homebrew.
 
 ```sh
-brew install ankane/brew/libtorch
+brew install libtorch
 ```
 
-Then install the gem (no need for `--with-torch-dir`).
+Then install the gem (no need for `bundle config`).
 
 ## rbenv
 

data/ext/torch/ext.cpp

@@ -441,19 +441,28 @@ void Init_ext()
         auto dtype = self.dtype();
 
         // TODO DRY if someone knows C++
-        // TODO kByte (uint8), kChar (int8), kBool (bool)
-        if (dtype == torch::kShort) {
-          short* data = self.data_ptr<short>();
+        if (dtype == torch::kByte) {
+          uint8_t* data = self.data_ptr<uint8_t>();
+          for (int i = 0; i < self.numel(); i++) {
+            a.push(data[i]);
+          }
+        } else if (dtype == torch::kChar) {
+          int8_t* data = self.data_ptr<int8_t>();
+          for (int i = 0; i < self.numel(); i++) {
+            a.push(to_ruby<int>(data[i]));
+          }
+        } else if (dtype == torch::kShort) {
+          int16_t* data = self.data_ptr<int16_t>();
           for (int i = 0; i < self.numel(); i++) {
             a.push(data[i]);
           }
         } else if (dtype == torch::kInt) {
-          int* data = self.data_ptr<int>();
+          int32_t* data = self.data_ptr<int32_t>();
           for (int i = 0; i < self.numel(); i++) {
             a.push(data[i]);
           }
         } else if (dtype == torch::kLong) {
-          long long* data = self.data_ptr<long long>();
+          int64_t* data = self.data_ptr<int64_t>();
           for (int i = 0; i < self.numel(); i++) {
             a.push(data[i]);
           }
@@ -467,8 +476,11 @@ void Init_ext()
           for (int i = 0; i < self.numel(); i++) {
             a.push(data[i]);
           }
+        } else if (dtype == torch::kBool) {
+          // bool
+          throw std::runtime_error("Type not supported yet");
         } else {
-          throw "Unsupported type";
+          throw std::runtime_error("Unsupported type");
         }
         return a;
       })
@@ -499,8 +511,11 @@ void Init_ext()
         torch::Layout l;
         if (layout == "strided") {
           l = torch::kStrided;
+        } else if (layout == "sparse") {
+          l = torch::kSparse;
+          throw std::runtime_error("Sparse layout not supported yet");
         } else {
-          throw "Unsupported layout";
+          throw std::runtime_error("Unsupported layout: " + layout);
         }
         return self.layout(l);
       })
@@ -513,7 +528,7 @@ void Init_ext()
         } else if (device == "cuda") {
           d = torch::kCUDA;
         } else {
-          throw "Unsupported device";
+          throw std::runtime_error("Unsupported device: " + device);
         }
         return self.device(d);
       })

data/ext/torch/extconf.rb

@@ -4,6 +4,9 @@ abort "Missing stdc++" unless have_library("stdc++")
 
 $CXXFLAGS << " -std=c++11"
 
+# needed for Linux pre-cxx11 ABI version
+# $CXXFLAGS << " -D_GLIBCXX_USE_CXX11_ABI=0"
+
 # silence ruby/intern.h warning
 $CXXFLAGS << " -Wno-deprecated-register"
 

data/lib/torch/version.rb

@@ -1,3 +1,3 @@
 module Torch
-  VERSION = "0.1.0"
+  VERSION = "0.1.1"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: torch-rb
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - Andrew Kane
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2019-11-26 00:00:00.000000000 Z
+date: 2019-11-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rice