types - Cudnn 是否支持 INT32 数据类型？

Question

设备：带有 cuda10 的 GeForce GTX 1080。正如参考所说，我设置CUDNN_DATA_INT32了aDesc，cDesc，输入数据都是int32。float 用于 HALF 和 FLOAT 张量，double 用于 DOUBLE 张量。. 但没有对 int 数据的描述。所以我尝试使用 int32 输入的 alpha 和 beta 的 int float 和 double 数据类型，都出现 cudnn 错误 。cudnn 是否支持 int 数据？或者我怎样才能以正确的方式使用它。 Error Number 9 CUDNN_STATUS_NOT_SUPPORTED

cudnnStatus_t cudnnAddTensor(
cudnnHandle_t handle,
const void *alpha,
const cudnnTensorDescriptor_t aDesc,
const void *A,
const void *beta,
const cudnnTensorDescriptor_t cDesc,
void *C)

i!     handle: type=cudnnHandle_t; streamId=0x3870eb0;
i!     alpha: type=CUDNN_DATA_FLOAT; val=0.000000;
i!     biasDesc: type=cudnnTensorDescriptor_t:
i!         dataType: type=cudnnDataType_t; val=CUDNN_DATA_INT32 (4);
i!         nbDims: type=int; val=4;
i!         dimA: type=int; val=[1,3,3,3];
i!         strideA: type=int; val=[27,9,3,1];
i!     biasData: location=dev; addr=0x10b4e6de400;
i!     beta: type=CUDNN_DATA_FLOAT; val=0.000000;
i!     srcDestDesc: type=cudnnTensorDescriptor_t:
i!         dataType: type=cudnnDataType_t; val=CUDNN_DATA_INT32 (4);
i!         nbDims: type=int; val=4;
i!         dimA: type=int; val=[1,3,3,3];
i!         strideA: type=int; val=[27,9,3,1];
i!     srcDestData: location=dev; addr=0x10b4e6de600;
i! Time: 2020-03-07T11:01:42.852528 (0d+0h+0m+3s since start)
i! Process=17985; Thread=17985; GPU=0; Handle=0x394c180; StreamId=0x3870eb0.

标签： typesintegercudnn