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
         
     |