Skip to content

Commit

Permalink
Rename computeType to dataType in cudnn interface
Browse files Browse the repository at this point in the history
Not sure why it's called compute type here, but the C-side of the API
also just calls it `dataType` (see [here](https://docs.nvidia.com/deeplearning/cudnn/api/index.html#cudnnGetConvolutionNdDescriptor)).
  • Loading branch information
RomeoV committed Jun 12, 2023
1 parent f74c546 commit a765151
Showing 1 changed file with 8 additions and 8 deletions.
16 changes: 8 additions & 8 deletions lib/cudnn/src/libcudnn.jl
Original file line number Diff line number Diff line change
Expand Up @@ -2927,51 +2927,51 @@ end
end

@checked function cudnnSetConvolution2dDescriptor(convDesc, pad_h, pad_w, u, v, dilation_h,
dilation_w, mode, computeType)
dilation_w, mode, dataType)
initialize_context()
@ccall libcudnn.cudnnSetConvolution2dDescriptor(convDesc::cudnnConvolutionDescriptor_t,
pad_h::Cint, pad_w::Cint, u::Cint,
v::Cint, dilation_h::Cint,
dilation_w::Cint,
mode::cudnnConvolutionMode_t,
computeType::cudnnDataType_t)::cudnnStatus_t
dataType::cudnnDataType_t)::cudnnStatus_t
end

@checked function cudnnGetConvolution2dDescriptor(convDesc, pad_h, pad_w, u, v, dilation_h,
dilation_w, mode, computeType)
dilation_w, mode, dataType)
initialize_context()
@ccall libcudnn.cudnnGetConvolution2dDescriptor(convDesc::cudnnConvolutionDescriptor_t,
pad_h::Ptr{Cint}, pad_w::Ptr{Cint},
u::Ptr{Cint}, v::Ptr{Cint},
dilation_h::Ptr{Cint},
dilation_w::Ptr{Cint},
mode::Ptr{cudnnConvolutionMode_t},
computeType::Ptr{cudnnDataType_t})::cudnnStatus_t
dataType::Ptr{cudnnDataType_t})::cudnnStatus_t
end

@checked function cudnnSetConvolutionNdDescriptor(convDesc, arrayLength, padA,
filterStrideA, dilationA, mode,
computeType)
dataType)
initialize_context()
@ccall libcudnn.cudnnSetConvolutionNdDescriptor(convDesc::cudnnConvolutionDescriptor_t,
arrayLength::Cint, padA::Ptr{Cint},
filterStrideA::Ptr{Cint},
dilationA::Ptr{Cint},
mode::cudnnConvolutionMode_t,
computeType::cudnnDataType_t)::cudnnStatus_t
dataType::cudnnDataType_t)::cudnnStatus_t
end

@checked function cudnnGetConvolutionNdDescriptor(convDesc, arrayLengthRequested,
arrayLength, padA, strideA, dilationA,
mode, computeType)
mode, dataType)
initialize_context()
@ccall libcudnn.cudnnGetConvolutionNdDescriptor(convDesc::cudnnConvolutionDescriptor_t,
arrayLengthRequested::Cint,
arrayLength::Ptr{Cint}, padA::Ptr{Cint},
strideA::Ptr{Cint},
dilationA::Ptr{Cint},
mode::Ptr{cudnnConvolutionMode_t},
computeType::Ptr{cudnnDataType_t})::cudnnStatus_t
dataType::Ptr{cudnnDataType_t})::cudnnStatus_t
end

@checked function cudnnGetConvolution2dForwardOutputDim(convDesc, inputTensorDesc,
Expand Down

0 comments on commit a765151

Please sign in to comment.