LOOP IVOPTS: Apply LEN_MASK_{LOAD,STORE}

Message ID 20230623140537.4156063-1-juzhe.zhong@rivai.ai
State Accepted
Headers
Series LOOP IVOPTS: Apply LEN_MASK_{LOAD,STORE} |

Checks

Context Check Description
snail/gcc-patch-check success Github commit url

Commit Message

juzhe.zhong@rivai.ai June 23, 2023, 2:05 p.m. UTC
  From: Ju-Zhe Zhong <juzhe.zhong@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(-)
  

Comments

Jeff Law June 23, 2023, 7:14 p.m. UTC | #1
On 6/23/23 08:05, juzhe.zhong@rivai.ai wrote:
> From: Ju-Zhe Zhong <juzhe.zhong@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
  
juzhe.zhong@rivai.ai June 23, 2023, 11:20 p.m. UTC | #2
Ok will send V2 soon.



juzhe.zhong@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.zhong@rivai.ai wrote:
> From: Ju-Zhe Zhong <juzhe.zhong@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
  

Patch

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))));
       return NULL_TREE;
 
     default: