Skip to content

Commit

Permalink
Merge branch 'smilib' of https://github.com/NWChemEx/TAMM into smilib
Browse files Browse the repository at this point in the history
  • Loading branch information
abagusetty committed Aug 21, 2024
2 parents e2d8532 + 2d1bdfe commit b22fa5a
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion src/tamm/execution_context.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@ ExecutionContext::ExecutionContext(ProcGroup pg, DistributionKind default_dist_k
#else
ranks_pn_ = GA_Cluster_nprocs(GA_Cluster_proc_nodeid(pg.rank().value()));
#endif
nnodes_ = pg.size().value() / ranks_pn_;
nnodes_ = pg.size().value() / ranks_pn_;
tamm::getHardwareGPUCount(&gpus_pn_);

#if defined(__APPLE__)
Expand Down
2 changes: 1 addition & 1 deletion src/tamm/gpu_streams.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -15,9 +15,9 @@
#include <cuda_runtime_api.h>
#include <nvml.h>
#elif defined(USE_HIP)
#include <rocm_smi/rocm_smi.h>
#include <hip/hip_runtime.h>
#include <rocblas/rocblas.h>
#include <rocm_smi/rocm_smi.h>
#elif defined(USE_DPCPP)
#include "sycl_device.hpp"
#include <oneapi/mkl/blas.hpp>
Expand Down

0 comments on commit b22fa5a

Please sign in to comment.