cumo 0.4.0 → 0.4.1
Sign up to get free protection for your applications and to get access to all the features.
- 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
|