From 413634aa8e27d4daed18d03e56da20046c62ce66 Mon Sep 17 00:00:00 2001 From: Boris Fomitchev Date: Thu, 12 Nov 2015 14:25:43 -0800 Subject: Natalia's fixed for BN. Added bntest.lua --- ffi.lua | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'ffi.lua') diff --git a/ffi.lua b/ffi.lua index c8ee963..61ac9ce 100644 --- a/ffi.lua +++ b/ffi.lua @@ -1072,7 +1072,7 @@ cudnnStatus_t cudnnBatchNormalizationForwardTraining( const cudnnTensorDescriptor_t xDesc, const void *x, /* NxCxHxW */ - /* const cudnnTensorDescriptor_t yDesc, */ + const cudnnTensorDescriptor_t yDesc, void *y, /* NxCxHxW */ /* Same shared desc for all the 6 tensors below in the argument list. */ @@ -1139,7 +1139,7 @@ cudnnStatus_t cudnnBatchNormalizationForwardInference( const cudnnTensorDescriptor_t xDesc, const void *x, /* NxCxHxW */ - /* const cudnnTensorDescriptor_t yDesc, */ + const cudnnTensorDescriptor_t yDesc, void *y, /* NxCxHxW */ /* Same desc for all 4 tensors below */ @@ -1188,9 +1188,9 @@ cudnnStatus_t cudnnBatchNormalizationBackward( const cudnnTensorDescriptor_t xDesc, /* same desc for x, dx, dy */ const void *x, - /* const cudnnTensorDescriptor_t dyDesc, */ + const cudnnTensorDescriptor_t dyDesc, const void *dy, - /* const cudnnTensorDescriptor_t dxDesc, */ + const cudnnTensorDescriptor_t dxDesc, void *dx, /* this tensor desc is used for all the 4 tensors below */ -- cgit v1.2.3