File tree Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -73,9 +73,9 @@ infiniopStatus_t cudaCreateAddDescriptor(CudaHandle_t handle,
7373}
7474
7575infiniopStatus_t cudaDestroyAddDescriptor (AddCudaDescriptor_t desc) {
76- cudaFree ((void *) desc->a_strides );
77- cudaFree ((void *) desc->b_strides );
78- cudaFree ((void *) desc->c_strides );
76+ checkCudaErrorWithCode ( cudaFree ((void *) desc->a_strides ), STATUS_EXECUTION_FAILED );
77+ checkCudaErrorWithCode ( cudaFree ((void *) desc->b_strides ), STATUS_EXECUTION_FAILED );
78+ checkCudaErrorWithCode ( cudaFree ((void *) desc->c_strides ), STATUS_EXECUTION_FAILED );
7979 delete desc;
8080 return STATUS_SUCCESS;
8181}
Original file line number Diff line number Diff line change @@ -45,7 +45,7 @@ infiniopStatus_t cudaCreateExpandDescriptor(CudaHandle_t handle,
4545}
4646
4747infiniopStatus_t cudaDestroyExpandDescriptor (ExpandCudaDescriptor_t desc) {
48- cudaFree ((void *) desc->strides_and_shape_d );
48+ checkCudaErrorWithCode ( cudaFree ((void *) desc->strides_and_shape_d ), STATUS_EXECUTION_FAILED );
4949 delete desc;
5050 return STATUS_SUCCESS;
5151}
You can’t perform that action at this time.
0 commit comments