kernel_format.h

// -*-c++-*-
#ifndef STORMM_KERNEL_MANAGER_H
#define STORMM_KERNEL_MANAGER_H
 
#include <map>
#include <string>
#ifdef STORMM_USE_HPC
#  ifdef STORMM_USE_CUDA
#    include <cuda_runtime.h>
#  endif
#endif
#include "copyright.h"
#include "DataTypes/stormm_vector_types.h"
#include "gpu_details.h"
 
namespace stormm {
namespace card {
 
#ifndef STORMM_USE_HPC
using data_types::int2;
#endif

const char generic_kernel_manager_name[] = "KernelManager";
  
class KernelFormat {
public:

  KernelFormat();
  
  KernelFormat(int lb_max_threads_per_block, int lb_min_blocks_per_smp, int register_usage_in,
               int shared_usage_in, int block_subdivision, const GpuDetails &gpu,
               const std::string &kernel_name_in = std::string(""));
 
  KernelFormat(int lb_max_threads_per_block, int lb_min_blocks_per_smp, int register_usage_in,
               int shared_usage_in, const GpuDetails &gpu,
               const std::string &kernel_name_in = std::string(""));
 
#ifdef STORMM_USE_HPC
#  ifdef STORMM_USE_CUDA
  KernelFormat(const cudaFuncAttributes &attr, int lb_min_blocks_per_smp, int block_subdivision,
               const GpuDetails &gpu, const std::string &kernel_name_in = std::string(""));
#  endif
#endif

  KernelFormat(const KernelFormat &original) = default;
  KernelFormat(KernelFormat &&original) = default;
  KernelFormat& operator=(const KernelFormat &other) = default;
  KernelFormat& operator=(KernelFormat &&other) = default;
  
  int2 getLaunchParameters() const;

  int getRegisterUsage() const;

  int getBlockSizeLimit() const;

  int getSharedMemoryRequirement() const;

  const std::string& getKernelName() const;
  
private:
  int block_size_limit;        
  int shared_usage;            
  int block_dimension;         
  int grid_dimension;          
  int register_usage;          
  std::string kernel_name;     
};

class KernelManager {
public:

  const GpuDetails& getGpu() const;

  void printLaunchParameters(const std::string &k_key = std::string(""),
                             const char* true_class_name = generic_kernel_manager_name) const;
 
protected:

  KernelManager(const GpuDetails &gpu_in = null_gpu);

  virtual ~KernelManager();

  GpuDetails gpu;

  std::map<std::string, KernelFormat> k_dictionary;
};
  
} // namespace card
} // namespace stormm
 
#endif