1 /* Tree-based target query functions relating to optabs 2 Copyright (C) 1987-2024 Free Software Foundation, Inc. 3 4 This file is part of GCC. 5 6 GCC is free software; you can redistribute it and/or modify it under 7 the terms of the GNU General Public License as published by the Free 8 Software Foundation; either version 3, or (at your option) any later 9 version. 10 11 GCC is distributed in the hope that it will be useful, but WITHOUT ANY 12 WARRANTY; without even the implied warranty of MERCHANTABILITY or 13 FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 14 for more details. 15 16 You should have received a copy of the GNU General Public License 17 along with GCC; see the file COPYING3. If not see 18 <http://www.gnu.org/licenses/>. */ 19 20 21 #include "config.h" 22 #include "system.h" 23 #include "coretypes.h" 24 #include "target.h" 25 #include "insn-codes.h" 26 #include "rtl.h" 27 #include "tree.h" 28 #include "memmodel.h" 29 #include "optabs.h" 30 #include "optabs-tree.h" 31 #include "stor-layout.h" 32 33 /* Return the optab used for computing the operation given by the tree code, 34 CODE and the tree EXP. This function is not always usable (for example, it 35 cannot give complete results for multiplication or division) but probably 36 ought to be relied on more widely throughout the expander. */ 37 optab 38 optab_for_tree_code (enum tree_code code, const_tree type, 39 enum optab_subtype subtype) 40 { 41 bool trapv; 42 switch (code) 43 { 44 case BIT_AND_EXPR: 45 return and_optab; 46 47 case BIT_IOR_EXPR: 48 return ior_optab; 49 50 case BIT_NOT_EXPR: 51 return one_cmpl_optab; 52 53 case BIT_XOR_EXPR: 54 return xor_optab; 55 56 case MULT_HIGHPART_EXPR: 57 return TYPE_UNSIGNED (type) ? umul_highpart_optab : smul_highpart_optab; 58 59 case CEIL_MOD_EXPR: 60 case FLOOR_MOD_EXPR: 61 case ROUND_MOD_EXPR: 62 /* {s,u}mod_optab implements TRUNC_MOD_EXPR. For scalar modes, 63 expansion has code to adjust TRUNC_MOD_EXPR into the desired other 64 modes, but for vector modes it does not. The adjustment code 65 should be instead emitted in tree-vect-patterns.cc. */ 66 if (VECTOR_TYPE_P (type)) 67 return unknown_optab; 68 /* FALLTHRU */ 69 case TRUNC_MOD_EXPR: 70 return TYPE_UNSIGNED (type) ? umod_optab : smod_optab; 71 72 case CEIL_DIV_EXPR: 73 case FLOOR_DIV_EXPR: 74 case ROUND_DIV_EXPR: 75 /* {,u}{s,u}div_optab implements {TRUNC,EXACT}_DIV_EXPR or RDIV_EXPR. 76 For scalar modes, expansion has code to adjust TRUNC_DIV_EXPR 77 into the desired other modes, but for vector modes it does not. 78 The adjustment code should be instead emitted in 79 tree-vect-patterns.cc. */ 80 if (VECTOR_TYPE_P (type)) 81 return unknown_optab; 82 /* FALLTHRU */ 83 case RDIV_EXPR: 84 case TRUNC_DIV_EXPR: 85 case EXACT_DIV_EXPR: 86 if (TYPE_SATURATING (type)) 87 return TYPE_UNSIGNED (type) ? usdiv_optab : ssdiv_optab; 88 return TYPE_UNSIGNED (type) ? udiv_optab : sdiv_optab; 89 90 case LSHIFT_EXPR: 91 if (VECTOR_TYPE_P (type)) 92 { 93 if (subtype == optab_vector) 94 return TYPE_SATURATING (type) ? unknown_optab : vashl_optab; 95 96 gcc_assert (subtype == optab_scalar); 97 } 98 if (TYPE_SATURATING (type)) 99 return TYPE_UNSIGNED (type) ? usashl_optab : ssashl_optab; 100 return ashl_optab; 101 102 case RSHIFT_EXPR: 103 if (VECTOR_TYPE_P (type)) 104 { 105 if (subtype == optab_vector) 106 return TYPE_UNSIGNED (type) ? vlshr_optab : vashr_optab; 107 108 gcc_assert (subtype == optab_scalar); 109 } 110 return TYPE_UNSIGNED (type) ? lshr_optab : ashr_optab; 111 112 case LROTATE_EXPR: 113 if (VECTOR_TYPE_P (type)) 114 { 115 if (subtype == optab_vector) 116 return vrotl_optab; 117 118 gcc_assert (subtype == optab_scalar); 119 } 120 return rotl_optab; 121 122 case RROTATE_EXPR: 123 if (VECTOR_TYPE_P (type)) 124 { 125 if (subtype == optab_vector) 126 return vrotr_optab; 127 128 gcc_assert (subtype == optab_scalar); 129 } 130 return rotr_optab; 131 132 case MAX_EXPR: 133 return TYPE_UNSIGNED (type) ? umax_optab : smax_optab; 134 135 case MIN_EXPR: 136 return TYPE_UNSIGNED (type) ? umin_optab : smin_optab; 137 138 case REALIGN_LOAD_EXPR: 139 return vec_realign_load_optab; 140 141 case WIDEN_SUM_EXPR: 142 return TYPE_UNSIGNED (type) ? usum_widen_optab : ssum_widen_optab; 143 144 case DOT_PROD_EXPR: 145 { 146 if (subtype == optab_vector_mixed_sign) 147 return usdot_prod_optab; 148 149 return (TYPE_UNSIGNED (type) ? udot_prod_optab : sdot_prod_optab); 150 } 151 152 case SAD_EXPR: 153 return TYPE_UNSIGNED (type) ? usad_optab : ssad_optab; 154 155 case WIDEN_MULT_PLUS_EXPR: 156 return (TYPE_UNSIGNED (type) 157 ? (TYPE_SATURATING (type) 158 ? usmadd_widen_optab : umadd_widen_optab) 159 : (TYPE_SATURATING (type) 160 ? ssmadd_widen_optab : smadd_widen_optab)); 161 162 case WIDEN_MULT_MINUS_EXPR: 163 return (TYPE_UNSIGNED (type) 164 ? (TYPE_SATURATING (type) 165 ? usmsub_widen_optab : umsub_widen_optab) 166 : (TYPE_SATURATING (type) 167 ? ssmsub_widen_optab : smsub_widen_optab)); 168 169 case VEC_WIDEN_MULT_HI_EXPR: 170 return (TYPE_UNSIGNED (type) 171 ? vec_widen_umult_hi_optab : vec_widen_smult_hi_optab); 172 173 case VEC_WIDEN_MULT_LO_EXPR: 174 return (TYPE_UNSIGNED (type) 175 ? vec_widen_umult_lo_optab : vec_widen_smult_lo_optab); 176 177 case VEC_WIDEN_MULT_EVEN_EXPR: 178 return (TYPE_UNSIGNED (type) 179 ? vec_widen_umult_even_optab : vec_widen_smult_even_optab); 180 181 case VEC_WIDEN_MULT_ODD_EXPR: 182 return (TYPE_UNSIGNED (type) 183 ? vec_widen_umult_odd_optab : vec_widen_smult_odd_optab); 184 185 case VEC_WIDEN_LSHIFT_HI_EXPR: 186 return (TYPE_UNSIGNED (type) 187 ? vec_widen_ushiftl_hi_optab : vec_widen_sshiftl_hi_optab); 188 189 case VEC_WIDEN_LSHIFT_LO_EXPR: 190 return (TYPE_UNSIGNED (type) 191 ? vec_widen_ushiftl_lo_optab : vec_widen_sshiftl_lo_optab); 192 193 case VEC_UNPACK_HI_EXPR: 194 return (TYPE_UNSIGNED (type) 195 ? vec_unpacku_hi_optab : vec_unpacks_hi_optab); 196 197 case VEC_UNPACK_LO_EXPR: 198 return (TYPE_UNSIGNED (type) 199 ? vec_unpacku_lo_optab : vec_unpacks_lo_optab); 200 201 case VEC_UNPACK_FLOAT_HI_EXPR: 202 /* The signedness is determined from input operand. */ 203 return (TYPE_UNSIGNED (type) 204 ? vec_unpacku_float_hi_optab : vec_unpacks_float_hi_optab); 205 206 case VEC_UNPACK_FLOAT_LO_EXPR: 207 /* The signedness is determined from input operand. */ 208 return (TYPE_UNSIGNED (type) 209 ? vec_unpacku_float_lo_optab : vec_unpacks_float_lo_optab); 210 211 case VEC_UNPACK_FIX_TRUNC_HI_EXPR: 212 /* The signedness is determined from output operand. */ 213 return (TYPE_UNSIGNED (type) 214 ? vec_unpack_ufix_trunc_hi_optab 215 : vec_unpack_sfix_trunc_hi_optab); 216 217 case VEC_UNPACK_FIX_TRUNC_LO_EXPR: 218 /* The signedness is determined from output operand. */ 219 return (TYPE_UNSIGNED (type) 220 ? vec_unpack_ufix_trunc_lo_optab 221 : vec_unpack_sfix_trunc_lo_optab); 222 223 case VEC_PACK_TRUNC_EXPR: 224 return vec_pack_trunc_optab; 225 226 case VEC_PACK_SAT_EXPR: 227 return TYPE_UNSIGNED (type) ? vec_pack_usat_optab : vec_pack_ssat_optab; 228 229 case VEC_PACK_FIX_TRUNC_EXPR: 230 /* The signedness is determined from output operand. */ 231 return (TYPE_UNSIGNED (type) 232 ? vec_pack_ufix_trunc_optab : vec_pack_sfix_trunc_optab); 233 234 case VEC_PACK_FLOAT_EXPR: 235 /* The signedness is determined from input operand. */ 236 return (TYPE_UNSIGNED (type) 237 ? vec_packu_float_optab : vec_packs_float_optab); 238 239 case VEC_DUPLICATE_EXPR: 240 return vec_duplicate_optab; 241 242 case VEC_SERIES_EXPR: 243 return vec_series_optab; 244 245 default: 246 break; 247 } 248 249 trapv = INTEGRAL_TYPE_P (type) && TYPE_OVERFLOW_TRAPS (type); 250 switch (code) 251 { 252 case POINTER_PLUS_EXPR: 253 case PLUS_EXPR: 254 if (TYPE_SATURATING (type)) 255 return TYPE_UNSIGNED (type) ? usadd_optab : ssadd_optab; 256 return trapv ? addv_optab : add_optab; 257 258 case POINTER_DIFF_EXPR: 259 case MINUS_EXPR: 260 if (TYPE_SATURATING (type)) 261 return TYPE_UNSIGNED (type) ? ussub_optab : sssub_optab; 262 return trapv ? subv_optab : sub_optab; 263 264 case MULT_EXPR: 265 if (TYPE_SATURATING (type)) 266 return TYPE_UNSIGNED (type) ? usmul_optab : ssmul_optab; 267 return trapv ? smulv_optab : smul_optab; 268 269 case NEGATE_EXPR: 270 if (TYPE_SATURATING (type)) 271 return TYPE_UNSIGNED (type) ? usneg_optab : ssneg_optab; 272 return trapv ? negv_optab : neg_optab; 273 274 case ABS_EXPR: 275 return trapv ? absv_optab : abs_optab; 276 277 case ABSU_EXPR: 278 return abs_optab; 279 default: 280 return unknown_optab; 281 } 282 } 283 284 /* Check whether an operation represented by CODE is a 'half' widening operation 285 in which the input vector type has half the number of bits of the output 286 vector type e.g. V8QI->V8HI. 287 288 This is handled by widening the inputs using NOP_EXPRs then using a 289 non-widening stmt e.g. MINUS_EXPR. RTL fusing converts these to the widening 290 hardware instructions if supported. 291 292 The more typical case (handled in supportable_widening_operation) is where 293 the input vector type has the same number of bits as the output vector type. 294 In this case half the elements of the input vectors must be processed at a 295 time into respective vector outputs with elements twice as wide i.e. a 296 'hi'/'lo' pair using codes such as VEC_WIDEN_MINUS_HI/LO. 297 298 Supported widening operations: 299 WIDEN_MULT_EXPR 300 WIDEN_LSHIFT_EXPR 301 302 Output: 303 - CODE1 - The non-widened code, which will be used after the inputs are 304 converted to the wide type. */ 305 bool 306 supportable_half_widening_operation (enum tree_code code, tree vectype_out, 307 tree vectype_in, enum tree_code *code1) 308 { 309 machine_mode m1,m2; 310 enum tree_code dummy_code; 311 optab op; 312 313 gcc_assert (VECTOR_TYPE_P (vectype_out) && VECTOR_TYPE_P (vectype_in)); 314 315 m1 = TYPE_MODE (vectype_out); 316 m2 = TYPE_MODE (vectype_in); 317 318 if (!VECTOR_MODE_P (m1) || !VECTOR_MODE_P (m2)) 319 return false; 320 321 if (maybe_ne (TYPE_VECTOR_SUBPARTS (vectype_in), 322 TYPE_VECTOR_SUBPARTS (vectype_out))) 323 return false; 324 325 switch (code) 326 { 327 case WIDEN_LSHIFT_EXPR: 328 *code1 = LSHIFT_EXPR; 329 break; 330 case WIDEN_MULT_EXPR: 331 *code1 = MULT_EXPR; 332 break; 333 default: 334 return false; 335 } 336 337 if (!supportable_convert_operation (NOP_EXPR, vectype_out, vectype_in, 338 &dummy_code)) 339 return false; 340 341 op = optab_for_tree_code (*code1, vectype_out, optab_vector); 342 return (optab_handler (op, TYPE_MODE (vectype_out)) != CODE_FOR_nothing); 343 } 344 345 /* Function supportable_convert_operation 346 347 Check whether an operation represented by the code CODE is a 348 convert operation that is supported by the target platform in 349 vector form (i.e., when operating on arguments of type VECTYPE_IN 350 producing a result of type VECTYPE_OUT). 351 352 Convert operations we currently support directly are FIX_TRUNC and FLOAT. 353 This function checks if these operations are supported 354 by the target platform directly (via vector tree-codes). 355 356 Output: 357 - CODE1 is code of vector operation to be used when 358 vectorizing the operation, if available. */ 359 360 bool 361 supportable_convert_operation (enum tree_code code, 362 tree vectype_out, tree vectype_in, 363 enum tree_code *code1) 364 { 365 machine_mode m1,m2; 366 bool truncp; 367 368 gcc_assert (VECTOR_TYPE_P (vectype_out) && VECTOR_TYPE_P (vectype_in)); 369 370 m1 = TYPE_MODE (vectype_out); 371 m2 = TYPE_MODE (vectype_in); 372 373 if (!VECTOR_MODE_P (m1) || !VECTOR_MODE_P (m2)) 374 return false; 375 376 /* First check if we can done conversion directly. */ 377 if ((code == FIX_TRUNC_EXPR 378 && can_fix_p (m1,m2,TYPE_UNSIGNED (vectype_out), &truncp) 379 != CODE_FOR_nothing) 380 || (code == FLOAT_EXPR 381 && can_float_p (m1,m2,TYPE_UNSIGNED (vectype_in)) 382 != CODE_FOR_nothing)) 383 { 384 *code1 = code; 385 return true; 386 } 387 388 if (GET_MODE_UNIT_PRECISION (m1) > GET_MODE_UNIT_PRECISION (m2) 389 && can_extend_p (m1, m2, TYPE_UNSIGNED (vectype_in))) 390 { 391 *code1 = code; 392 return true; 393 } 394 395 if (GET_MODE_UNIT_PRECISION (m1) < GET_MODE_UNIT_PRECISION (m2) 396 && convert_optab_handler (trunc_optab, m1, m2) != CODE_FOR_nothing) 397 { 398 *code1 = code; 399 return true; 400 } 401 402 return false; 403 } 404 405 /* Return true iff vec_cmp_optab/vec_cmpu_optab can handle a vector comparison 406 for code CODE, comparing operands of type VALUE_TYPE and producing a result 407 of type MASK_TYPE. */ 408 409 static bool 410 vec_cmp_icode_p (tree value_type, tree mask_type, enum tree_code code) 411 { 412 enum rtx_code rcode = get_rtx_code_1 (code, TYPE_UNSIGNED (value_type)); 413 if (rcode == UNKNOWN) 414 return false; 415 416 return can_vec_cmp_compare_p (rcode, TYPE_MODE (value_type), 417 TYPE_MODE (mask_type)); 418 } 419 420 /* Return true iff vec_cmpeq_optab can handle a vector comparison for code 421 CODE, comparing operands of type VALUE_TYPE and producing a result of type 422 MASK_TYPE. */ 423 424 static bool 425 vec_cmp_eq_icode_p (tree value_type, tree mask_type, enum tree_code code) 426 { 427 if (code != EQ_EXPR && code != NE_EXPR) 428 return false; 429 430 return get_vec_cmp_eq_icode (TYPE_MODE (value_type), TYPE_MODE (mask_type)) 431 != CODE_FOR_nothing; 432 } 433 434 /* Return TRUE if appropriate vector insn is available 435 for vector comparison expr with vector type VALUE_TYPE 436 and resulting mask with MASK_TYPE. */ 437 438 bool 439 expand_vec_cmp_expr_p (tree value_type, tree mask_type, enum tree_code code) 440 { 441 return vec_cmp_icode_p (value_type, mask_type, code) 442 || vec_cmp_eq_icode_p (value_type, mask_type, code); 443 } 444 445 /* Return true iff vcond_optab/vcondu_optab can handle a vector 446 comparison for code CODE, comparing operands of type CMP_OP_TYPE and 447 producing a result of type VALUE_TYPE. */ 448 449 static bool 450 vcond_icode_p (tree value_type, tree cmp_op_type, enum tree_code code) 451 { 452 enum rtx_code rcode = get_rtx_code_1 (code, TYPE_UNSIGNED (cmp_op_type)); 453 if (rcode == UNKNOWN) 454 return false; 455 456 return can_vcond_compare_p (rcode, TYPE_MODE (value_type), 457 TYPE_MODE (cmp_op_type)); 458 } 459 460 /* Return true iff vcondeq_optab can handle a vector comparison for code CODE, 461 comparing operands of type CMP_OP_TYPE and producing a result of type 462 VALUE_TYPE. */ 463 464 static bool 465 vcond_eq_icode_p (tree value_type, tree cmp_op_type, enum tree_code code) 466 { 467 if (code != EQ_EXPR && code != NE_EXPR) 468 return false; 469 470 return get_vcond_eq_icode (TYPE_MODE (value_type), TYPE_MODE (cmp_op_type)) 471 != CODE_FOR_nothing; 472 } 473 474 /* Return TRUE iff, appropriate vector insns are available 475 for vector cond expr with vector type VALUE_TYPE and a comparison 476 with operand vector types in CMP_OP_TYPE. */ 477 478 bool 479 expand_vec_cond_expr_p (tree value_type, tree cmp_op_type, enum tree_code code) 480 { 481 machine_mode value_mode = TYPE_MODE (value_type); 482 machine_mode cmp_op_mode = TYPE_MODE (cmp_op_type); 483 if (VECTOR_BOOLEAN_TYPE_P (cmp_op_type) 484 && get_vcond_mask_icode (TYPE_MODE (value_type), 485 TYPE_MODE (cmp_op_type)) != CODE_FOR_nothing) 486 return true; 487 488 if (maybe_ne (GET_MODE_NUNITS (value_mode), GET_MODE_NUNITS (cmp_op_mode))) 489 return false; 490 491 if (TREE_CODE_CLASS (code) != tcc_comparison) 492 /* This may happen, for example, if code == SSA_NAME, in which case we 493 cannot be certain whether a vector insn is available. */ 494 return false; 495 496 return vcond_icode_p (value_type, cmp_op_type, code) 497 || vcond_eq_icode_p (value_type, cmp_op_type, code); 498 } 499 500 /* Use the current target and options to initialize 501 TREE_OPTIMIZATION_OPTABS (OPTNODE). */ 502 503 void 504 init_tree_optimization_optabs (tree optnode) 505 { 506 /* Quick exit if we have already computed optabs for this target. */ 507 if (TREE_OPTIMIZATION_BASE_OPTABS (optnode) == this_target_optabs) 508 return; 509 510 /* Forget any previous information and set up for the current target. */ 511 TREE_OPTIMIZATION_BASE_OPTABS (optnode) = this_target_optabs; 512 struct target_optabs *tmp_optabs = (struct target_optabs *) 513 TREE_OPTIMIZATION_OPTABS (optnode); 514 if (tmp_optabs) 515 memset (tmp_optabs, 0, sizeof (struct target_optabs)); 516 else 517 tmp_optabs = ggc_cleared_alloc<target_optabs> (); 518 519 /* Generate a new set of optabs into tmp_optabs. */ 520 init_all_optabs (tmp_optabs); 521 522 /* If the optabs changed, record it. */ 523 if (memcmp (tmp_optabs, this_target_optabs, sizeof (struct target_optabs))) 524 TREE_OPTIMIZATION_OPTABS (optnode) = tmp_optabs; 525 else 526 { 527 TREE_OPTIMIZATION_OPTABS (optnode) = NULL; 528 ggc_free (tmp_optabs); 529 } 530 } 531 532 /* Return TRUE if the target has support for vector right shift of an 533 operand of type TYPE. If OT_TYPE is OPTAB_DEFAULT, check for existence 534 of a shift by either a scalar or a vector. Otherwise, check only 535 for a shift that matches OT_TYPE. */ 536 537 bool 538 target_supports_op_p (tree type, enum tree_code code, 539 enum optab_subtype ot_subtype) 540 { 541 optab ot = optab_for_tree_code (code, type, ot_subtype); 542 return (ot != unknown_optab 543 && optab_handler (ot, TYPE_MODE (type)) != CODE_FOR_nothing); 544 } 545 546 /* Return true if the target has support for masked load/store. 547 We can support masked load/store by either mask{load,store} 548 or mask_len_{load,store}. 549 This helper function checks whether target supports masked 550 load/store and return corresponding IFN in the last argument 551 (IFN_MASK_{LOAD,STORE} or IFN_MASK_LEN_{LOAD,STORE}). */ 552 553 static bool 554 target_supports_mask_load_store_p (machine_mode mode, machine_mode mask_mode, 555 bool is_load, internal_fn *ifn) 556 { 557 optab op = is_load ? maskload_optab : maskstore_optab; 558 optab len_op = is_load ? mask_len_load_optab : mask_len_store_optab; 559 if (convert_optab_handler (op, mode, mask_mode) != CODE_FOR_nothing) 560 { 561 if (ifn) 562 *ifn = is_load ? IFN_MASK_LOAD : IFN_MASK_STORE; 563 return true; 564 } 565 else if (convert_optab_handler (len_op, mode, mask_mode) != CODE_FOR_nothing) 566 { 567 if (ifn) 568 *ifn = is_load ? IFN_MASK_LEN_LOAD : IFN_MASK_LEN_STORE; 569 return true; 570 } 571 return false; 572 } 573 574 /* Return true if target supports vector masked load/store for mode. 575 An additional output in the last argument which is the IFN pointer. 576 We set IFN as MASK_{LOAD,STORE} or MASK_LEN_{LOAD,STORE} according 577 which optab is supported in the target. */ 578 579 bool 580 can_vec_mask_load_store_p (machine_mode mode, 581 machine_mode mask_mode, 582 bool is_load, 583 internal_fn *ifn) 584 { 585 machine_mode vmode; 586 587 /* If mode is vector mode, check it directly. */ 588 if (VECTOR_MODE_P (mode)) 589 return target_supports_mask_load_store_p (mode, mask_mode, is_load, ifn); 590 591 /* Otherwise, return true if there is some vector mode with 592 the mask load/store supported. */ 593 594 /* See if there is any chance the mask load or store might be 595 vectorized. If not, punt. */ 596 scalar_mode smode; 597 if (!is_a <scalar_mode> (mode, &smode)) 598 return false; 599 600 vmode = targetm.vectorize.preferred_simd_mode (smode); 601 if (VECTOR_MODE_P (vmode) 602 && targetm.vectorize.get_mask_mode (vmode).exists (&mask_mode) 603 && target_supports_mask_load_store_p (vmode, mask_mode, is_load, ifn)) 604 return true; 605 606 auto_vector_modes vector_modes; 607 targetm.vectorize.autovectorize_vector_modes (&vector_modes, true); 608 for (machine_mode base_mode : vector_modes) 609 if (related_vector_mode (base_mode, smode).exists (&vmode) 610 && targetm.vectorize.get_mask_mode (vmode).exists (&mask_mode) 611 && target_supports_mask_load_store_p (vmode, mask_mode, is_load, ifn)) 612 return true; 613 return false; 614 } 615 616 /* Return true if the target has support for len load/store. 617 We can support len load/store by either len_{load,store} 618 or mask_len_{load,store}. 619 This helper function checks whether target supports len 620 load/store and return corresponding IFN in the last argument 621 (IFN_LEN_{LOAD,STORE} or IFN_MASK_LEN_{LOAD,STORE}). */ 622 623 static bool 624 target_supports_len_load_store_p (machine_mode mode, bool is_load, 625 internal_fn *ifn) 626 { 627 optab op = is_load ? len_load_optab : len_store_optab; 628 optab masked_op = is_load ? mask_len_load_optab : mask_len_store_optab; 629 630 if (direct_optab_handler (op, mode)) 631 { 632 if (ifn) 633 *ifn = is_load ? IFN_LEN_LOAD : IFN_LEN_STORE; 634 return true; 635 } 636 machine_mode mask_mode; 637 if (targetm.vectorize.get_mask_mode (mode).exists (&mask_mode) 638 && convert_optab_handler (masked_op, mode, mask_mode) != CODE_FOR_nothing) 639 { 640 if (ifn) 641 *ifn = is_load ? IFN_MASK_LEN_LOAD : IFN_MASK_LEN_STORE; 642 return true; 643 } 644 return false; 645 } 646 647 /* If target supports vector load/store with length for vector mode MODE, 648 return the corresponding vector mode, otherwise return opt_machine_mode (). 649 There are two flavors for vector load/store with length, one is to measure 650 length with bytes, the other is to measure length with lanes. 651 As len_{load,store} optabs point out, for the flavor with bytes, we use 652 VnQI to wrap the other supportable same size vector modes. 653 An additional output in the last argument which is the IFN pointer. 654 We set IFN as LEN_{LOAD,STORE} or MASK_LEN_{LOAD,STORE} according 655 which optab is supported in the target. */ 656 657 opt_machine_mode 658 get_len_load_store_mode (machine_mode mode, bool is_load, internal_fn *ifn) 659 { 660 gcc_assert (VECTOR_MODE_P (mode)); 661 662 /* Check if length in lanes supported for this mode directly. */ 663 if (target_supports_len_load_store_p (mode, is_load, ifn)) 664 return mode; 665 666 /* Check if length in bytes supported for same vector size VnQI. */ 667 machine_mode vmode; 668 poly_uint64 nunits = GET_MODE_SIZE (mode); 669 if (related_vector_mode (mode, QImode, nunits).exists (&vmode) 670 && target_supports_len_load_store_p (vmode, is_load, ifn)) 671 return vmode; 672 673 return opt_machine_mode (); 674 } 675