Skip to content

Commit 573ba28

Browse files
hdelanhdelan
committed
Respond to comments
- Check that TensorDim < 3 using yaml returns: . - Rename some things and remove copypasta
1 parent 72935f1 commit 573ba28

File tree

9 files changed

+74
-48
lines changed

9 files changed

+74
-48
lines changed

‎include/ur_api.h

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -9585,7 +9585,7 @@ urEnqueueNativeCommandExp(
95859585
#if !defined(__GNUC__)
95869586
#pragma endregion
95879587
#endif
9588-
// Intel 'oneAPI' Unified Runtime Experimental API for enqueuing work through native APIs
9588+
// Intel 'oneAPI' Unified Runtime Experimental API for mapping tensor objects
95899589
#if !defined(__GNUC__)
95909590
#pragma region tensor map(experimental)
95919591
#endif
@@ -9706,6 +9706,8 @@ typedef enum ur_exp_tensor_map_oob_fill_flag_t {
97069706
/// + `NULL == PixelBoxUpperCorner`
97079707
/// + `NULL == ElementStrides`
97089708
/// + `NULL == hTensorMap`
9709+
/// - ::UR_RESULT_ERROR_INVALID_ARGUMENT
9710+
/// + `TensorRank < 3`
97099711
UR_APIEXPORT ur_result_t UR_APICALL
97109712
urTensorMapEncodeIm2ColExp(
97119713
ur_device_handle_t hDevice, ///< [in] Handle of the device object.
@@ -9715,12 +9717,12 @@ urTensorMapEncodeIm2ColExp(
97159717
const uint64_t *GlobalDim, ///< [in] Array containing tensor size (number of elements) along each of
97169718
///< the TensorRank dimensions.
97179719
const uint64_t *GlobalStrides, ///< [in] Array containing stride size (in bytes) along each of the
9718-
///< tensorRank - 1 dimensions.
9720+
///< TensorRank - 1 dimensions.
97199721
const int *PixelBoxLowerCorner, ///< [in] Array containing DHW dimensions of lower box corner.
97209722
const int *PixelBoxUpperCorner, ///< [in] Array containing DHW dimensions of upper box corner.
97219723
uint32_t ChannelsPerPixel, ///< [in] Number of channels per pixel.
97229724
uint32_t PixelsPerColumn, ///< [in] Number of pixels per column.
9723-
const uint32_t *ElementStrides, ///< [in] Array containing traversal stride in each of the tensorRank
9725+
const uint32_t *ElementStrides, ///< [in] Array containing traversal stride in each of the TensorRank
97249726
///< dimensions.
97259727
ur_exp_tensor_map_interleave_flags_t Interleave, ///< [in] Type of interleaved layout the tensor addresses
97269728
ur_exp_tensor_map_swizzle_flags_t Swizzle, ///< [in] Bank swizzling pattern inside shared memory
@@ -9756,6 +9758,8 @@ urTensorMapEncodeIm2ColExp(
97569758
/// + `NULL == BoxDim`
97579759
/// + `NULL == ElementStrides`
97589760
/// + `NULL == hTensorMap`
9761+
/// - ::UR_RESULT_ERROR_INVALID_ARGUMENT
9762+
/// + `TensorRank < 3`
97599763
UR_APIEXPORT ur_result_t UR_APICALL
97609764
urTensorMapEncodeTiledExp(
97619765
ur_device_handle_t hDevice, ///< [in] Handle of the device object.
@@ -9765,11 +9769,11 @@ urTensorMapEncodeTiledExp(
97659769
const uint64_t *GlobalDim, ///< [in] Array containing tensor size (number of elements) along each of
97669770
///< the TensorRank dimensions.
97679771
const uint64_t *GlobalStrides, ///< [in] Array containing stride size (in bytes) along each of the
9768-
///< tensorRank - 1 dimensions.
9772+
///< TensorRank - 1 dimensions.
97699773
const uint32_t *BoxDim, ///< [in] Array containing traversal box size (number of elments) along
9770-
///< each of the tensorRank dimensions. Specifies how many elements to be
9774+
///< each of the TensorRank dimensions. Specifies how many elements to be
97719775
///< traversed along each tensor dimension.
9772-
const uint32_t *ElementStrides, ///< [in] Array containing traversal stride in each of the tensorRank
9776+
const uint32_t *ElementStrides, ///< [in] Array containing traversal stride in each of the TensorRank
97739777
///< dimensions.
97749778
ur_exp_tensor_map_interleave_flags_t Interleave, ///< [in] Type of interleaved layout the tensor addresses
97759779
ur_exp_tensor_map_swizzle_flags_t Swizzle, ///< [in] Bank swizzling pattern inside shared memory

‎scripts/core/exp-tensor-map.yml

Lines changed: 12 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
#
1010
--- #--------------------------------------------------------------------------
1111
type: header
12-
desc: "Intel $OneApi Unified Runtime Experimental API for enqueuing work through native APIs"
12+
desc: "Intel $OneApi Unified Runtime Experimental API for mapping tensor objects"
1313
ordinal: "100"
1414

1515
--- #--------------------------------------------------------------------------
@@ -125,7 +125,7 @@ params:
125125
desc: "[in] Array containing tensor size (number of elements) along each of the TensorRank dimensions."
126126
- type: const uint64_t*
127127
name: GlobalStrides
128-
desc: "[in] Array containing stride size (in bytes) along each of the tensorRank - 1 dimensions."
128+
desc: "[in] Array containing stride size (in bytes) along each of the TensorRank - 1 dimensions."
129129
- type: const int*
130130
name: PixelBoxLowerCorner
131131
desc: "[in] Array containing DHW dimensions of lower box corner."
@@ -140,7 +140,7 @@ params:
140140
desc: "[in] Number of pixels per column."
141141
- type: const uint32_t*
142142
name: ElementStrides
143-
desc: "[in] Array containing traversal stride in each of the tensorRank dimensions."
143+
desc: "[in] Array containing traversal stride in each of the TensorRank dimensions."
144144
- type: $x_exp_tensor_map_interleave_flags_t
145145
name: Interleave
146146
desc: "[in] Type of interleaved layout the tensor addresses"
@@ -156,6 +156,9 @@ params:
156156
- type: $x_exp_tensor_map_handle_t*
157157
name: hTensorMap
158158
desc: "[out] Handle of the tensor map object."
159+
returns:
160+
- $X_RESULT_ERROR_INVALID_ARGUMENT:
161+
- "`TensorRank < 3`"
159162

160163
--- #--------------------------------------------------------------------------
161164
type: function
@@ -182,13 +185,13 @@ params:
182185
desc: "[in] Array containing tensor size (number of elements) along each of the TensorRank dimensions."
183186
- type: const uint64_t*
184187
name: GlobalStrides
185-
desc: "[in] Array containing stride size (in bytes) along each of the tensorRank - 1 dimensions."
188+
desc: "[in] Array containing stride size (in bytes) along each of the TensorRank - 1 dimensions."
186189
- type: const uint32_t*
187190
name: BoxDim
188-
desc: "[in] Array containing traversal box size (number of elments) along each of the tensorRank dimensions. Specifies how many elements to be traversed along each tensor dimension."
191+
desc: "[in] Array containing traversal box size (number of elments) along each of the TensorRank dimensions. Specifies how many elements to be traversed along each tensor dimension."
189192
- type: const uint32_t*
190193
name: ElementStrides
191-
desc: "[in] Array containing traversal stride in each of the tensorRank dimensions."
194+
desc: "[in] Array containing traversal stride in each of the TensorRank dimensions."
192195
- type: $x_exp_tensor_map_interleave_flags_t
193196
name: Interleave
194197
desc: "[in] Type of interleaved layout the tensor addresses"
@@ -204,4 +207,7 @@ params:
204207
- type: $x_exp_tensor_map_handle_t*
205208
name: hTensorMap
206209
desc: "[out] Handle of the tensor map object."
210+
returns:
211+
- $X_RESULT_ERROR_INVALID_ARGUMENT:
212+
- "`TensorRank < 3`"
207213

‎source/adapters/cuda/tensor_map.cpp

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -18,8 +18,8 @@ struct ur_exp_tensor_map_handle_t_ {
1818
};
1919

2020
#define CONVERT(URTYPE, CUTYPE) \
21-
if (URTYPE & UrType) \
22-
return CUTYPE;
21+
if ((URTYPE)&UrType) \
22+
return (CUTYPE);
2323

2424
inline CUtensorMapDataType
2525
convertUrToCuDataType(ur_exp_tensor_map_data_type_flags_t UrType) {
@@ -71,7 +71,7 @@ convertUrToCuSwizzle(ur_exp_tensor_map_swizzle_flags_t UrType) {
7171
}
7272

7373
CUtensorMapL2promotion
74-
convertUrToL2promotion(ur_exp_tensor_map_l2_promotion_flags_t UrType) {
74+
convertUrToCuL2Promotion(ur_exp_tensor_map_l2_promotion_flags_t UrType) {
7575
CONVERT(UR_EXP_TENSOR_MAP_L2_PROMOTION_FLAG_NONE,
7676
CU_TENSOR_MAP_L2_PROMOTION_NONE);
7777
CONVERT(UR_EXP_TENSOR_MAP_L2_PROMOTION_FLAG_64B,
@@ -84,7 +84,7 @@ convertUrToL2promotion(ur_exp_tensor_map_l2_promotion_flags_t UrType) {
8484
}
8585

8686
CUtensorMapFloatOOBfill
87-
convertUrToCuOOBfill(ur_exp_tensor_map_oob_fill_flags_t UrType) {
87+
convertUrToCuOobFill(ur_exp_tensor_map_oob_fill_flags_t UrType) {
8888
CONVERT(UR_EXP_TENSOR_MAP_OOB_FILL_FLAG_NONE,
8989
CU_TENSOR_MAP_FLOAT_OOB_FILL_NONE);
9090
CONVERT(UR_EXP_TENSOR_MAP_OOB_FILL_FLAG_REQUEST_ZERO_FMA,
@@ -111,7 +111,7 @@ UR_APIEXPORT ur_result_t UR_APICALL urTensorMapEncodeIm2ColExp(
111111
GlobalAddress, GlobalDim, GlobalStrides, PixelBoxLowerCorner,
112112
PixelBoxUpperCorner, ChannelsPerPixel, PixelsPerColumn, ElementStrides,
113113
convertUrToCuInterleave(Interleave), convertUrToCuSwizzle(Swizzle),
114-
convertUrToL2promotion(L2Promotion), convertUrToCuOOBfill(OobFill)));
114+
convertUrToCuL2Promotion(L2Promotion), convertUrToCuOobFill(OobFill)));
115115
} catch (ur_result_t Err) {
116116
return Err;
117117
}
@@ -134,7 +134,7 @@ UR_APIEXPORT ur_result_t UR_APICALL urTensorMapEncodeTiledExp(
134134
&(*hTensorMap)->Map, convertUrToCuDataType(TensorMapType), TensorRank,
135135
GlobalAddress, GlobalDim, GlobalStrides, BoxDim, ElementStrides,
136136
convertUrToCuInterleave(Interleave), convertUrToCuSwizzle(Swizzle),
137-
convertUrToL2promotion(L2Promotion), convertUrToCuOOBfill(OobFill)));
137+
convertUrToCuL2Promotion(L2Promotion), convertUrToCuOobFill(OobFill)));
138138
} catch (ur_result_t Err) {
139139
return Err;
140140
}

‎source/adapters/null/ur_nullddi.cpp

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -5919,15 +5919,15 @@ __urdlllocal ur_result_t UR_APICALL urTensorMapEncodeIm2ColExp(
59195919
///< the TensorRank dimensions.
59205920
const uint64_t *
59215921
GlobalStrides, ///< [in] Array containing stride size (in bytes) along each of the
5922-
///< tensorRank - 1 dimensions.
5922+
///< TensorRank - 1 dimensions.
59235923
const int *
59245924
PixelBoxLowerCorner, ///< [in] Array containing DHW dimensions of lower box corner.
59255925
const int *
59265926
PixelBoxUpperCorner, ///< [in] Array containing DHW dimensions of upper box corner.
59275927
uint32_t ChannelsPerPixel, ///< [in] Number of channels per pixel.
59285928
uint32_t PixelsPerColumn, ///< [in] Number of pixels per column.
59295929
const uint32_t *
5930-
ElementStrides, ///< [in] Array containing traversal stride in each of the tensorRank
5930+
ElementStrides, ///< [in] Array containing traversal stride in each of the TensorRank
59315931
///< dimensions.
59325932
ur_exp_tensor_map_interleave_flags_t
59335933
Interleave, ///< [in] Type of interleaved layout the tensor addresses
@@ -5977,13 +5977,13 @@ __urdlllocal ur_result_t UR_APICALL urTensorMapEncodeTiledExp(
59775977
///< the TensorRank dimensions.
59785978
const uint64_t *
59795979
GlobalStrides, ///< [in] Array containing stride size (in bytes) along each of the
5980-
///< tensorRank - 1 dimensions.
5980+
///< TensorRank - 1 dimensions.
59815981
const uint32_t *
59825982
BoxDim, ///< [in] Array containing traversal box size (number of elments) along
5983-
///< each of the tensorRank dimensions. Specifies how many elements to be
5983+
///< each of the TensorRank dimensions. Specifies how many elements to be
59845984
///< traversed along each tensor dimension.
59855985
const uint32_t *
5986-
ElementStrides, ///< [in] Array containing traversal stride in each of the tensorRank
5986+
ElementStrides, ///< [in] Array containing traversal stride in each of the TensorRank
59875987
///< dimensions.
59885988
ur_exp_tensor_map_interleave_flags_t
59895989
Interleave, ///< [in] Type of interleaved layout the tensor addresses

‎source/loader/layers/tracing/ur_trcddi.cpp

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -7921,15 +7921,15 @@ __urdlllocal ur_result_t UR_APICALL urTensorMapEncodeIm2ColExp(
79217921
///< the TensorRank dimensions.
79227922
const uint64_t *
79237923
GlobalStrides, ///< [in] Array containing stride size (in bytes) along each of the
7924-
///< tensorRank - 1 dimensions.
7924+
///< TensorRank - 1 dimensions.
79257925
const int *
79267926
PixelBoxLowerCorner, ///< [in] Array containing DHW dimensions of lower box corner.
79277927
const int *
79287928
PixelBoxUpperCorner, ///< [in] Array containing DHW dimensions of upper box corner.
79297929
uint32_t ChannelsPerPixel, ///< [in] Number of channels per pixel.
79307930
uint32_t PixelsPerColumn, ///< [in] Number of pixels per column.
79317931
const uint32_t *
7932-
ElementStrides, ///< [in] Array containing traversal stride in each of the tensorRank
7932+
ElementStrides, ///< [in] Array containing traversal stride in each of the TensorRank
79337933
///< dimensions.
79347934
ur_exp_tensor_map_interleave_flags_t
79357935
Interleave, ///< [in] Type of interleaved layout the tensor addresses
@@ -8004,13 +8004,13 @@ __urdlllocal ur_result_t UR_APICALL urTensorMapEncodeTiledExp(
80048004
///< the TensorRank dimensions.
80058005
const uint64_t *
80068006
GlobalStrides, ///< [in] Array containing stride size (in bytes) along each of the
8007-
///< tensorRank - 1 dimensions.
8007+
///< TensorRank - 1 dimensions.
80088008
const uint32_t *
80098009
BoxDim, ///< [in] Array containing traversal box size (number of elments) along
8010-
///< each of the tensorRank dimensions. Specifies how many elements to be
8010+
///< each of the TensorRank dimensions. Specifies how many elements to be
80118011
///< traversed along each tensor dimension.
80128012
const uint32_t *
8013-
ElementStrides, ///< [in] Array containing traversal stride in each of the tensorRank
8013+
ElementStrides, ///< [in] Array containing traversal stride in each of the TensorRank
80148014
///< dimensions.
80158015
ur_exp_tensor_map_interleave_flags_t
80168016
Interleave, ///< [in] Type of interleaved layout the tensor addresses

‎source/loader/layers/validation/ur_valddi.cpp

Lines changed: 13 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -9597,15 +9597,15 @@ __urdlllocal ur_result_t UR_APICALL urTensorMapEncodeIm2ColExp(
95979597
///< the TensorRank dimensions.
95989598
const uint64_t *
95999599
GlobalStrides, ///< [in] Array containing stride size (in bytes) along each of the
9600-
///< tensorRank - 1 dimensions.
9600+
///< TensorRank - 1 dimensions.
96019601
const int *
96029602
PixelBoxLowerCorner, ///< [in] Array containing DHW dimensions of lower box corner.
96039603
const int *
96049604
PixelBoxUpperCorner, ///< [in] Array containing DHW dimensions of upper box corner.
96059605
uint32_t ChannelsPerPixel, ///< [in] Number of channels per pixel.
96069606
uint32_t PixelsPerColumn, ///< [in] Number of pixels per column.
96079607
const uint32_t *
9608-
ElementStrides, ///< [in] Array containing traversal stride in each of the tensorRank
9608+
ElementStrides, ///< [in] Array containing traversal stride in each of the TensorRank
96099609
///< dimensions.
96109610
ur_exp_tensor_map_interleave_flags_t
96119611
Interleave, ///< [in] Type of interleaved layout the tensor addresses
@@ -9678,6 +9678,10 @@ __urdlllocal ur_result_t UR_APICALL urTensorMapEncodeIm2ColExp(
96789678
if (UR_EXP_TENSOR_MAP_OOB_FILL_FLAGS_MASK & OobFill) {
96799679
return UR_RESULT_ERROR_INVALID_ENUMERATION;
96809680
}
9681+
9682+
if (TensorRank < 3) {
9683+
return UR_RESULT_ERROR_INVALID_ARGUMENT;
9684+
}
96819685
}
96829686

96839687
if (context.enableLifetimeValidation &&
@@ -9708,13 +9712,13 @@ __urdlllocal ur_result_t UR_APICALL urTensorMapEncodeTiledExp(
97089712
///< the TensorRank dimensions.
97099713
const uint64_t *
97109714
GlobalStrides, ///< [in] Array containing stride size (in bytes) along each of the
9711-
///< tensorRank - 1 dimensions.
9715+
///< TensorRank - 1 dimensions.
97129716
const uint32_t *
97139717
BoxDim, ///< [in] Array containing traversal box size (number of elments) along
9714-
///< each of the tensorRank dimensions. Specifies how many elements to be
9718+
///< each of the TensorRank dimensions. Specifies how many elements to be
97159719
///< traversed along each tensor dimension.
97169720
const uint32_t *
9717-
ElementStrides, ///< [in] Array containing traversal stride in each of the tensorRank
9721+
ElementStrides, ///< [in] Array containing traversal stride in each of the TensorRank
97189722
///< dimensions.
97199723
ur_exp_tensor_map_interleave_flags_t
97209724
Interleave, ///< [in] Type of interleaved layout the tensor addresses
@@ -9782,6 +9786,10 @@ __urdlllocal ur_result_t UR_APICALL urTensorMapEncodeTiledExp(
97829786
if (UR_EXP_TENSOR_MAP_OOB_FILL_FLAGS_MASK & OobFill) {
97839787
return UR_RESULT_ERROR_INVALID_ENUMERATION;
97849788
}
9789+
9790+
if (TensorRank < 3) {
9791+
return UR_RESULT_ERROR_INVALID_ARGUMENT;
9792+
}
97859793
}
97869794

97879795
if (context.enableLifetimeValidation &&

‎source/loader/ur_ldrddi.cpp

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -8186,15 +8186,15 @@ __urdlllocal ur_result_t UR_APICALL urTensorMapEncodeIm2ColExp(
81868186
///< the TensorRank dimensions.
81878187
const uint64_t *
81888188
GlobalStrides, ///< [in] Array containing stride size (in bytes) along each of the
8189-
///< tensorRank - 1 dimensions.
8189+
///< TensorRank - 1 dimensions.
81908190
const int *
81918191
PixelBoxLowerCorner, ///< [in] Array containing DHW dimensions of lower box corner.
81928192
const int *
81938193
PixelBoxUpperCorner, ///< [in] Array containing DHW dimensions of upper box corner.
81948194
uint32_t ChannelsPerPixel, ///< [in] Number of channels per pixel.
81958195
uint32_t PixelsPerColumn, ///< [in] Number of pixels per column.
81968196
const uint32_t *
8197-
ElementStrides, ///< [in] Array containing traversal stride in each of the tensorRank
8197+
ElementStrides, ///< [in] Array containing traversal stride in each of the TensorRank
81988198
///< dimensions.
81998199
ur_exp_tensor_map_interleave_flags_t
82008200
Interleave, ///< [in] Type of interleaved layout the tensor addresses
@@ -8256,13 +8256,13 @@ __urdlllocal ur_result_t UR_APICALL urTensorMapEncodeTiledExp(
82568256
///< the TensorRank dimensions.
82578257
const uint64_t *
82588258
GlobalStrides, ///< [in] Array containing stride size (in bytes) along each of the
8259-
///< tensorRank - 1 dimensions.
8259+
///< TensorRank - 1 dimensions.
82608260
const uint32_t *
82618261
BoxDim, ///< [in] Array containing traversal box size (number of elments) along
8262-
///< each of the tensorRank dimensions. Specifies how many elements to be
8262+
///< each of the TensorRank dimensions. Specifies how many elements to be
82638263
///< traversed along each tensor dimension.
82648264
const uint32_t *
8265-
ElementStrides, ///< [in] Array containing traversal stride in each of the tensorRank
8265+
ElementStrides, ///< [in] Array containing traversal stride in each of the TensorRank
82668266
///< dimensions.
82678267
ur_exp_tensor_map_interleave_flags_t
82688268
Interleave, ///< [in] Type of interleaved layout the tensor addresses

0 commit comments

Comments
 (0)