Skip to content

Commit d269360

Browse files
hdelanHugh Delaney
andauthored
[SYCL][CUDA] ifdef cuda 11.5 datatypes (#10691)
These new datatypes were introduced in CUDA 11.5 Co-authored-by: Hugh Delaney <[email protected]>
1 parent 25cec26 commit d269360

File tree

1 file changed

+4
-1
lines changed
  • sycl/plugins/unified_runtime/ur/adapters/cuda

1 file changed

+4
-1
lines changed

sycl/plugins/unified_runtime/ur/adapters/cuda/image.cpp

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -71,10 +71,13 @@ urToCudaImageChannelFormat(ur_image_channel_type_t image_channel_type,
7171
PixelTypeSizeBytes = SIZE; \
7272
break; \
7373
}
74+
// These new formats were brought in in CUDA 11.5
75+
#if CUDA_VERSION >= 11050
7476
CASE(UR_IMAGE_CHANNEL_TYPE_UNORM_INT8, CU_AD_FORMAT_UNORM_INT8X1, 1)
77+
CASE(UR_IMAGE_CHANNEL_TYPE_UNORM_INT16, CU_AD_FORMAT_UNORM_INT16X1, 2)
78+
#endif
7579
CASE(UR_IMAGE_CHANNEL_TYPE_UNSIGNED_INT8, CU_AD_FORMAT_UNSIGNED_INT8, 1)
7680
CASE(UR_IMAGE_CHANNEL_TYPE_SIGNED_INT8, CU_AD_FORMAT_SIGNED_INT8, 1)
77-
CASE(UR_IMAGE_CHANNEL_TYPE_UNORM_INT16, CU_AD_FORMAT_UNORM_INT16X1, 2)
7881
CASE(UR_IMAGE_CHANNEL_TYPE_UNSIGNED_INT16, CU_AD_FORMAT_UNSIGNED_INT16, 2)
7982
CASE(UR_IMAGE_CHANNEL_TYPE_SIGNED_INT16, CU_AD_FORMAT_SIGNED_INT16, 2)
8083
CASE(UR_IMAGE_CHANNEL_TYPE_HALF_FLOAT, CU_AD_FORMAT_HALF, 2)

0 commit comments

Comments
 (0)