Ok will send V2 soon.
juzhe.zh...@rivai.ai From: Jeff Law Date: 2023-06-24 03:14 To: juzhe.zhong; gcc-patches CC: rguenther; richard.sandiford Subject: Re: [PATCH] LOOP IVOPTS: Apply LEN_MASK_{LOAD,STORE} On 6/23/23 08:05, juzhe.zh...@rivai.ai wrote: > From: Ju-Zhe Zhong <juzhe.zh...@rivai.ai> > > gcc/ChangeLog: > > * tree-ssa-loop-ivopts.cc (get_mem_type_for_internal_fn): Apply > LEN_MASK_{LOAD,STORE}. > > --- > gcc/tree-ssa-loop-ivopts.cc | 6 +++++- > 1 file changed, 5 insertions(+), 1 deletion(-) > > diff --git a/gcc/tree-ssa-loop-ivopts.cc b/gcc/tree-ssa-loop-ivopts.cc > index 6671ff6db5a..2b66fe66bc7 100644 > --- a/gcc/tree-ssa-loop-ivopts.cc > +++ b/gcc/tree-ssa-loop-ivopts.cc > @@ -2442,6 +2442,7 @@ get_mem_type_for_internal_fn (gcall *call, tree *op_p) > case IFN_MASK_LOAD: > case IFN_MASK_LOAD_LANES: > case IFN_LEN_LOAD: > + case IFN_LEN_MASK_LOAD: > if (op_p == gimple_call_arg_ptr (call, 0)) > return TREE_TYPE (gimple_call_lhs (call)); > return NULL_TREE; > @@ -2449,8 +2450,11 @@ get_mem_type_for_internal_fn (gcall *call, tree *op_p) > case IFN_MASK_STORE: > case IFN_MASK_STORE_LANES: > case IFN_LEN_STORE: > + case IFN_LEN_MASK_STORE: > if (op_p == gimple_call_arg_ptr (call, 0)) > - return TREE_TYPE (gimple_call_arg (call, 3)); > + return TREE_TYPE ( > + gimple_call_arg (call, internal_fn_stored_value_index ( > + gimple_call_internal_fn (call)))); Formatting nit. Compute the result of internal_fn_stored_value_index into a temporary and pass that temporary into gimple_call_arg which should clean up the formatting here. In general, if you find yourself indenting after an open paren like you've done here, compute the value into a temporary. OK with the formatting fix. jeff