/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck/utility/scheduler_enum.hpp Source File

/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck/utility/scheduler_enum.hpp Source File#

Composable Kernel: /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck/utility/scheduler_enum.hpp Source File

Go to the documentation of this file.

 // Copyright (c) Advanced Micro Devices, Inc., or its affiliates.
 // SPDX-License-Identifier: MIT
  
 #pragma once
  
 #if !defined(__HIPCC_RTC__) || !defined(CK_CODE_GEN_RTC)
 #include <ostream>
 #endif
  
 namespace ck {
  
 enum struct BlockGemmPipelineVersion
 {
     // For GEMM
     v1, 
     v2, 
     v3, 
     v4, 
     v5, 
  
     // For GEMM with preshuffled weight
     // v1, single lds buffer
     // v2, double lds buffer
 };
  
 enum struct BlockGemmPipelineScheduler
 {
     Intrawave, 
     Interwave, 
 };
  
 enum struct LoopScheduler
 {
     Default,   
     Interwave, 
 };
  
 enum struct TailNumber
 {
     // Single / Double buffer pipeline
     Odd,  
     Even, 
  
     // Long prefetch pipeline, up to 8
     One,   
     Two,   
     Three, 
     Four,  
     Five,  
     Six,   
     Seven, 
  
     // Unroll stages > Prefetch stages, number of loop is multiple of unroll stages
     Empty, 
            // Unroll stages <= Prefetch stages, number of loop is multiple of unroll stages add
            // prefetchstages
     Full,  
 };
  
 } // namespace ck
  
 #if !defined(__HIPCC_RTC__) || !defined(CK_CODE_GEN_RTC)
 inline std::ostream& operator<<(std::ostream& os, const ck::LoopScheduler& s)
 {
     switch(s)
     {
     case ck::LoopScheduler::Default: os << "Default"; break;
     case ck::LoopScheduler::Interwave: os << "Interwave"; break;
     default: os << "";
     }
     return os;
 }
 #endif