src/hotspot/share/opto/chaitin.hpp

rev 60615 : 8231441: Initial SVE backend support
Reviewed-by: adinn, pli
Contributed-by: joshua.zhu@arm.com, yang.zhang@arm.com, ningsheng.jian@arm.com

*** 1,7 ****
  /*
!  * Copyright (c) 1997, 2019, Oracle and/or its affiliates. All rights reserved.
   * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   *
   * This code is free software; you can redistribute it and/or modify it
   * under the terms of the GNU General Public License version 2 only, as
   * published by the Free Software Foundation.
--- 1,7 ----
  /*
!  * Copyright (c) 1997, 2020, Oracle and/or its affiliates. All rights reserved.
   * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   *
   * This code is free software; you can redistribute it and/or modify it
   * under the terms of the GNU General Public License version 2 only, as
   * published by the Free Software Foundation.
*** 112,122 ****
--- 112,124 ----
      _mask_size = size;
  #ifdef ASSERT
      _msize_valid=1;
      if (_is_vector) {
        assert(!_fat_proj, "sanity");
+       if (!(_is_scalable && OptoReg::is_stack(_reg))) {
          assert(_mask.is_aligned_sets(_num_regs), "mask is not aligned, adjacent sets");
+       }
      } else if (_num_regs == 2 && !_fat_proj) {
        assert(_mask.is_aligned_pairs(), "mask is not aligned, adjacent pairs");
      }
  #endif
    }
*** 135,152 ****
  
    void Insert( OptoReg::Name reg ) { _mask.Insert(reg);  debug_only(_msize_valid=0;) }
    void Remove( OptoReg::Name reg ) { _mask.Remove(reg);  debug_only(_msize_valid=0;) }
    void clear_to_sets()  { _mask.clear_to_sets(_num_regs); debug_only(_msize_valid=0;) }
  
-   // Number of registers this live range uses when it colors
  private:
    uint16_t _num_regs;           // 2 for Longs and Doubles, 1 for all else
                                  // except _num_regs is kill count for fat_proj
  public:
    int num_regs() const { return _num_regs; }
    void set_num_regs( int reg ) { assert( _num_regs == reg || !_num_regs, "" ); _num_regs = reg; }
  
  private:
    // Number of physical registers this live range uses when it colors
    // Architecture and register-set dependent
    uint16_t _reg_pressure;
  public:
--- 137,177 ----
  
    void Insert( OptoReg::Name reg ) { _mask.Insert(reg);  debug_only(_msize_valid=0;) }
    void Remove( OptoReg::Name reg ) { _mask.Remove(reg);  debug_only(_msize_valid=0;) }
    void clear_to_sets()  { _mask.clear_to_sets(_num_regs); debug_only(_msize_valid=0;) }
  
  private:
+   // Number of registers this live range uses when it colors
    uint16_t _num_regs;           // 2 for Longs and Doubles, 1 for all else
                                  // except _num_regs is kill count for fat_proj
+ 
+   // For scalable register, num_regs may not be the actual physical register size.
+   // We need to get the actual physical length of scalable register when scalable
+   // register is spilled. The size of one slot is 32-bit.
+   uint _scalable_reg_slots;     // Actual scalable register length of slots.
+                                 // Meaningful only when _is_scalable is true.
  public:
    int num_regs() const { return _num_regs; }
    void set_num_regs( int reg ) { assert( _num_regs == reg || !_num_regs, "" ); _num_regs = reg; }
  
+   uint scalable_reg_slots() { return _scalable_reg_slots; }
+   void set_scalable_reg_slots(uint slots) {
+     assert(_is_scalable, "scalable register");
+     assert(slots > 0, "slots of scalable register is not valid");
+     _scalable_reg_slots = slots;
+   }
+ 
+   bool is_scalable() {
+ #ifdef ASSERT
+     if (_is_scalable) {
+       // Should only be a vector for now, but it could also be a RegVMask in future.
+       assert(_is_vector && (_num_regs == RegMask::SlotsPerVecA), "unexpected scalable reg");
+     }
+ #endif
+     return _is_scalable;
+   }
+ 
  private:
    // Number of physical registers this live range uses when it colors
    // Architecture and register-set dependent
    uint16_t _reg_pressure;
  public:
*** 168,177 ****
--- 193,204 ----
    bool just_lo_degree () const { return degree() == degrees_of_freedom(); }
  
    uint   _is_oop:1,             // Live-range holds an oop
           _is_float:1,           // True if in float registers
           _is_vector:1,          // True if in vector registers
+          _is_scalable:1,        // True if register size is scalable
+                                 //      e.g. Arm SVE vector/predicate registers.
           _was_spilled1:1,       // True if prior spilling on def
           _was_spilled2:1,       // True if twice prior spilling on def
           _is_bound:1,           // live range starts life with no
                                  // degrees of freedom.
           _direct_conflict:1,    // True if def and use registers in conflict