|
13 | 13 |
|
14 | 14 | #include "context.hpp"
|
15 | 15 |
|
| 16 | +#if CUDA_VERSION < 12000 |
| 17 | +UR_APIEXPORT ur_result_t UR_APICALL urTensorMapEncodeIm2ColExp( |
| 18 | + ur_device_handle_t, ur_exp_tensor_map_data_type_flags_t, uint32_t, void *, |
| 19 | + const uint64_t *, const uint64_t *, const int *, const int *, uint32_t, |
| 20 | + uint32_t, const uint32_t *, ur_exp_tensor_map_interleave_flags_t, |
| 21 | + ur_exp_tensor_map_swizzle_flags_t, ur_exp_tensor_map_l2_promotion_flags_t, |
| 22 | + ur_exp_tensor_map_oob_fill_flags_t, ur_exp_tensor_map_handle_t *) { |
| 23 | + return UR_RESULT_ERROR_UNSUPPORTED_FEATURE; |
| 24 | +} |
| 25 | +UR_APIEXPORT ur_result_t UR_APICALL urTensorMapEncodeTiledExp( |
| 26 | + ur_device_handle_t, ur_exp_tensor_map_data_type_flags_t, uint32_t, void *, |
| 27 | + const uint64_t *, const uint64_t *, const uint32_t *, const uint32_t *, |
| 28 | + ur_exp_tensor_map_interleave_flags_t, ur_exp_tensor_map_swizzle_flags_t, |
| 29 | + ur_exp_tensor_map_l2_promotion_flags_t, ur_exp_tensor_map_oob_fill_flags_t, |
| 30 | + ur_exp_tensor_map_handle_t *) { |
| 31 | + return UR_RESULT_ERROR_UNSUPPORTED_FEATURE; |
| 32 | +} |
| 33 | +#else |
16 | 34 | struct ur_exp_tensor_map_handle_t_ {
|
17 | 35 | CUtensorMap Map;
|
18 | 36 | };
|
@@ -140,3 +158,4 @@ UR_APIEXPORT ur_result_t UR_APICALL urTensorMapEncodeTiledExp(
|
140 | 158 | }
|
141 | 159 | return UR_RESULT_SUCCESS;
|
142 | 160 | }
|
| 161 | +#endif |
0 commit comments