Поддерживает ли Cudnn тип данных INT32? - PullRequest
Новая
       10

Поддерживает ли Cudnn тип данных INT32?

0 голосов
/ 07 марта 2020

Устройство: GeForce GTX 1080 с cuda10. как говорится в ссылке, я установил CUDNN_DATA_INT32 для aDes c, cDes c, а все входные данные - int32. с плавающей запятой для тезисов HALF и FLOAT и удвоенные для тензоров DOUBLE. . но нет описания для данных int. поэтому я попробовал int float и double datatype для alpha и beta с вводом int32, все приходит с ошибкой cudnn Error Number 9 CUDNN_STATUS_NOT_SUPPORTED. Cudnn поддерживает int данные? или как я могу использовать это правильно. 

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.

Спасибо !!!

...