cumo 0.4.0 → 0.4.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- checksums.yaml +4 -4
- data/CHANGELOG.md +6 -0
- data/README.md +1 -1
- data/ext/cumo/include/cumo.h +2 -2
- data/ext/cumo/narray/gen/tmpl/fixed_batch_norm.c +4 -4
- metadata +2 -2
checksums.yaml
CHANGED
@@ -1,7 +1,7 @@
|
|
1
1
|
---
|
2
2
|
SHA256:
|
3
|
-
metadata.gz:
|
4
|
-
data.tar.gz:
|
3
|
+
metadata.gz: 784e23c2ba8be26ed4a9834766f4a410b6a73fe2b828a696b3f8107bdd187861
|
4
|
+
data.tar.gz: 9ff0de0b005907b070868cf40548b3b6c883459898e83b252784a7e30dc77058
|
5
5
|
SHA512:
|
6
|
-
metadata.gz:
|
7
|
-
data.tar.gz:
|
6
|
+
metadata.gz: 6cb2f338da7f52501e0d3d6716990a9ddc4165b2aa564b5e22afe98526a151a25474f27081850719fc00e49c0407859af8470070666de0653e4cd74bf0155866
|
7
|
+
data.tar.gz: 3a4362323eb239376644bfb49857afce658d79f8e7ec786e993502592387caa281c9a626fae01b374f28ab6ae8dcb4bc435c7fe73f97728b3200b9f39409b357
|
data/CHANGELOG.md
CHANGED
data/README.md
CHANGED
@@ -7,7 +7,7 @@ Cumo (pronounced "koomo") is a CUDA-aware, GPU-optimized numerical library that
|
|
7
7
|
## Requirements
|
8
8
|
|
9
9
|
* Ruby 2.5 or later
|
10
|
-
* NVIDIA GPU Compute Capability
|
10
|
+
* NVIDIA GPU Compute Capability 3.5 (Kepler) or later
|
11
11
|
* CUDA 9.0 or later
|
12
12
|
|
13
13
|
## Preparation
|
data/ext/cumo/include/cumo.h
CHANGED
@@ -104,11 +104,11 @@ static VALUE
|
|
104
104
|
y_ptr = cumo_na_get_offset_pointer_for_write(y);
|
105
105
|
|
106
106
|
status = cumo_cuda_cudnn_CreateTensorDescriptor(&x_desc, x_cont, cudnn_dtype);
|
107
|
-
if (status != CUDNN_STATUS_SUCCESS) goto
|
107
|
+
if (status != CUDNN_STATUS_SUCCESS) goto FIXED_BATCH_NORM_ERROR;
|
108
108
|
|
109
109
|
mode = cumo_cuda_cudnn_GetBatchNormMode(axis_ndim, int_axis);
|
110
110
|
status = cumo_cuda_cudnn_CreateBNTensorDescriptor(&bn_desc, x_desc, mode);
|
111
|
-
if (status != CUDNN_STATUS_SUCCESS) goto
|
111
|
+
if (status != CUDNN_STATUS_SUCCESS) goto FIXED_BATCH_NORM_ERROR;
|
112
112
|
// TODO: bn_desc may return another type, and may need to cast gamma, beta, mean, var
|
113
113
|
|
114
114
|
handle = cumo_cuda_cudnn_handle();
|
@@ -128,9 +128,9 @@ static VALUE
|
|
128
128
|
mean_cont_ptr,
|
129
129
|
var_cont_ptr,
|
130
130
|
double_eps);
|
131
|
-
if (status != CUDNN_STATUS_SUCCESS) goto
|
131
|
+
if (status != CUDNN_STATUS_SUCCESS) goto FIXED_BATCH_NORM_ERROR;
|
132
132
|
|
133
|
-
|
133
|
+
FIXED_BATCH_NORM_ERROR:
|
134
134
|
if (x_desc) cudnnDestroyTensorDescriptor(x_desc);
|
135
135
|
if (bn_desc) cudnnDestroyTensorDescriptor(bn_desc);
|
136
136
|
cumo_cuda_cudnn_check_status(status);
|
metadata
CHANGED
@@ -1,14 +1,14 @@
|
|
1
1
|
--- !ruby/object:Gem::Specification
|
2
2
|
name: cumo
|
3
3
|
version: !ruby/object:Gem::Version
|
4
|
-
version: 0.4.
|
4
|
+
version: 0.4.1
|
5
5
|
platform: ruby
|
6
6
|
authors:
|
7
7
|
- Naotoshi Seo
|
8
8
|
autorequire:
|
9
9
|
bindir: exe
|
10
10
|
cert_chain: []
|
11
|
-
date: 2019-05-
|
11
|
+
date: 2019-05-06 00:00:00.000000000 Z
|
12
12
|
dependencies:
|
13
13
|
- !ruby/object:Gem::Dependency
|
14
14
|
name: numo-narray
|