EigenExpValue copy.cpp File Reference

Functions
int	main (int argc, char **argv)

Variables
namespace	filesystem = std::experimental::filesystem

Function Documentation

main()

int main	(	int	argc,
		char **	argv
	)

                                {
  #include "PBC_TI/Framework/Fragments/setVariablesForEnsemble.cpp"
 
  Integer_t dim_sub; //全ヒルベルト空間の次元
  Integer_t info, failed=0; // カウンタ
  double gE, EnergyRange, OpRange, OpMin;
 
  if( !Initialize(argc, argv, Nargs_common) ) {
    std::cerr << "Error: Initialization failed." << std::endl;
    std::exit(EX_USAGE);
  }
  debug_print("# Successfully initialized.");
 
  //******************** Check for the directory structure ********************
  debug_print("# Checking for the directory structure.");
  std::ofstream OutFs;
  //******************** (END) Check for the directory structure ********************
 
  //***************************************************************************
  //******************** Allocation & Initialization **************************
  //***************************************************************************
#ifdef GPU
  magma_init();
  magma_queue_t queue = NULL;
  magma_int_t     dev = 0;
  magma_getdevice( &dev );
  magma_queue_create( dev, &queue );
#else
  void* GPUconf = nullptr;
#endif
 
  //******************** Translation invariance ********************
  debug_print("# Calculating translation-invariant sectors.");
  #include "PBC_TI/Framework/Fragments/TranslationInvariantSectors.cpp"
  //******************** (END)Translation invariance ********************
 
  //********** Allocate CPU memories **********//
  debug_print("# Allocating CPU memories.");
  Integer_t const dim_max = SectorDimension(Sector[n_max]);
  std::vector<Integer_t>   NdataInShell(dim_max);
  std::vector<double>         eigenEnergy(dim_max);
  std::vector<double>            EXPvalue(dim_max);
  matrix<Complex_t>        h(dloc_h , dloc_h );
  matrix<Complex_t>      loc(dloc_op, dloc_op);
#ifndef GPU
  matrix<Complex_t>    h_tot(dim_max, dim_max);
  matrix<Complex_t>  loc_tot(dim_max, dim_max);
  #define   dh   h
  #define dloc loc
  #define   dh_tot   h_tot
  #define dloc_tot loc_tot
#endif
  //********** (END) Allocate CPU memories **********//
 
#ifdef GPU
  //********** Allocate GPU memories **********//
  debug_print("# Allocating GPU memories.");
  constexpr Integer_t GPU_UNIT = 32;
  Integer_t const LDT = magma_roundup(dim_max, GPU_UNIT);
  matrix_gpu<Complex_t>       dh(dloc_h , dloc_h );
  matrix_gpu<Complex_t>     dloc(dloc_op, dloc_op);
  matrix_gpu<Complex_t>   dh_tot(LDT, dim_max);
  matrix_gpu<Complex_t> dloc_tot(LDT, dim_max);
  //********** (END) Allocate GPU memories **********//
 
  //********** Determine GPU configuration **********//
  #include "PBC_TI/Framework/Fragments/getAttributesOfMatrixElementsInSector.cpp"
  //********** (END) Determine GPU configuration **********//
#endif // #ifdef GPU
 
  double start, t_int, end, temp_t;
  double T_diag=0, T_post=0, T_pre=0;
  init_genrand(SEED);
  start = getETtime();
  for(Integer_t repetition = 0;repetition < repMin; ++repetition) {
    generateLocal_h( h,   dloc_h,  -1);
    generateLocal_op(loc, dloc_op, -1);
  }
  end = getETtime();
  std::cout << "(init_genrand): time=" << std::fixed << (end-start) << std::endl;
  //***************************************************************************
  //******************** (END) Allocation & Initialization ********************
  //***************************************************************************
 
  start = getETtime();
  end   = start;
  for(Integer_t repetition = repMin;repetition <= repMax; ++repetition) {
    // 局所ハミルトニアンと局所物理量をランダムにとる ******************************//
    generateLocal_h( h,   dloc_h,  -1);
    generateLocal_op(loc, dloc_op, -1);
      h.print(dloc_h, dloc_h);
    loc.print(dloc_op, dloc_op);
 
    debug_print("# Seting matrix to GPU.");
  #ifdef GPU
    magma_setmatrix(dloc_h,  dloc_h,  sizeof(Complex_t), &*h.begin(),   dloc_h,  dh.ptr(),   dloc_h,  queue);
    magma_setmatrix(dloc_op, dloc_op, sizeof(Complex_t), &*loc.begin(), dloc_op, dloc.ptr(), dloc_op, queue);
  #endif
    debug_print("# Set matrix to GPU.");
 
    std::string outDirName(baseDirName);
    {
      std::stringstream buff;
      buff << "/RawData/Sample_No" << repetition;
      outDirName += buff.str();
      outDirName = std::regex_replace(outDirName, std::regex("//"), "/");
      filesystem::create_directories(outDirName);
    }
 
    for(size_t n = n_max;n >= n_min; --n) {
      debug_print("# (rep,n)=(" << repetition << "," << n << ")");
      debug_print("# Constructing global matrices in the sector.");
      temp_t  = getETtime();
      {
        dim_sub = constructGlobal_h(   dh_tot, dh,   num_h,  Sector.at(n),GPUconf);
                  constructGlobal_op(dloc_tot, dloc, num_op, Sector.at(n),GPUconf);
        #ifdef GPU
          magma_queue_sync(queue);
        #endif
      }
      T_pre += getETtime() -temp_t;
 
      // magma_queue_sync(queue);
      // std::cerr << "(N=" << n << ") isnan_kernel(dh_tot)   before" << std::endl;
      //   isnan_kernel<<<conf.dimGrid(),conf.dimBlock(),conf.shared(),conf.stream()>>>(dim_sub, dh_tot.ptr(), dh_tot.LD());
      // magma_queue_sync(queue);
      // std::cerr << "(N=" << n << ") isnan_kernel(dloc_tot) before" << std::endl;
      //   isnan_kernel<<<conf.dimGrid(),conf.dimBlock(),conf.shared(),conf.stream()>>>(dim_sub, dloc_tot.ptr(), dloc_tot.LD());
      // magma_queue_sync(queue);
 
      temp_t = getETtime();
      {
        debug_print("# Calculating eigenstate expectation values.");
        info = EigenExpValue(EXPvalue, eigenEnergy, dim_sub, dh_tot, dloc_tot,GPUconf);
          if(info != 0) {
            #include "PBC_TI/Framework/Fragments/writeFailedMatrixToFile.cpp"
            continue;
          }
        EnergyRange = eigenEnergy[dim_sub-1] - eigenEnergy[0];
        gE = eigenEnergy[0];
        debug_print("# Calculating the spectral range of the observable.");
        OpRange = SpectralRange(OpMin, dim_sub, dloc_tot);
      }
      T_diag += getETtime() -temp_t;
 
      // magma_queue_sync(queue);
      // std::cerr << "(N=" << n << ") isnan_kernel(dh_tot)   after" << std::endl;
      //   isnan_kernel<<<conf.dimGrid(),conf.dimBlock(),conf.shared(),conf.stream()>>>(dim_sub, dh_tot.ptr(), dh_tot.LD());
      // magma_queue_sync(queue);
      // std::cerr << "(N=" << n << ") isnan_kernel(dloc_tot) after" << std::endl;
      //   isnan_kernel<<<conf.dimGrid(),conf.dimBlock(),conf.shared(),conf.stream()>>>(dim_sub, dloc_tot.ptr(), dloc_tot.LD());
      // magma_queue_sync(queue);
 
      temp_t  = getETtime();
      {
        debug_print("# Writing results to a file.");
        std::stringstream buff("");
        buff << "/EigenExpValue" << PRECISION << "_N" << n << ".txt";
        std::string filename(outDirName); filename += buff.str();
        OutFs.open(filename); checkIsFileOpen(OutFs, filename);
        OutFs << std::right << std::showpos << std::scientific << std::setprecision(6);
        OutFs << "# energyRange= " << EnergyRange << "\n"
              << "# gE= "          << gE          << "\n"
              << "# OpRange= "     << OpRange     << "\n"
              << "# OpMin= "       << OpMin       << "\n"
              << "# 1.(Normalized energy) 2.(Energy) 3.(Exp value)" << "\n\n";
        for(size_t j = 0;j < dim_sub; ++j) {
          OutFs << (eigenEnergy[j] - gE)/EnergyRange << " "
                << eigenEnergy[j]                    << " "
                << EXPvalue[j] << std::endl;
        }
        OutFs.close();
      }
      T_post += getETtime() -temp_t;
    }
    if(repetition%10 == 9) {
      t_int = end; end = getETtime();
      std::cerr << "(total=" << std::setw(6) << repetition+1
                << "): timeINT="    << std::setprecision(6)  << std::setw(8)  << (end-t_int)
                << ", timeTOT="     << std::setprecision(6)  << std::setw(8)  << (end-start)
                << ", T_construct=" << std::setprecision(6)  << std::setw(10) << T_pre  << "(" << std::setprecision(1) << 100*T_pre /(end-start) << "%)"
                << ", T_diag="      << std::setprecision(6)  << std::setw(8)  << T_diag << "(" << std::setprecision(1) << 100*T_diag/(end-start) << "%)"
                << ", T_process="   << std::setprecision(6)  << std::setw(8)  << T_post << "(" << std::setprecision(1) << 100*T_post/(end-start) << "%)"
                << std::endl;
    }
  }
 
  Finalize(argc, argv);
  #ifdef GPU
    magma_finalize();
  #endif
  return 0;
}

Variable Documentation

filesystem

namespace filesystem = std::experimental::filesystem