device_array_ops.h -> device_array.h

Author: ajay-mk

examples/device/ta_dense_um_tensor.cpp

@@ -20,6 +20,7 @@
 // clang-format off
 #include <tiledarray.h>
 #include <TiledArray/device/um_tensor.h>
+#include <TiledArray/device/device_array.h>
 // clang-format on
 
 #ifdef TILEDARRAY_HAS_CUDA

src/CMakeLists.txt

@@ -235,7 +235,7 @@ if(TILEDARRAY_HAS_HIP OR TILEDARRAY_HAS_CUDA)
           TiledArray/device/btas.h
           TiledArray/device/btas_um_tensor.h
           TiledArray/device/um_tensor.h
-          TiledArray/device/device_array_ops.h
+          TiledArray/device/device_array.h
           TiledArray/device/device_task_fn.h
           TiledArray/device/kernel/mult_kernel.h
           TiledArray/device/kernel/reduce_kernel.h

src/TiledArray/device/btas_um_tensor.h

@@ -34,7 +34,7 @@
 
 #include <TiledArray/device/blas.h>
 #include <TiledArray/device/btas.h>
-#include <TiledArray/device/device_array_ops.h>
+#include <TiledArray/device/device_array.h>
 #include <TiledArray/device/um_storage.h>
 #include <TiledArray/external/librett.h>
 #include <TiledArray/tile.h>

src/TiledArray/device/device_array_ops.h src/TiledArray/device/device_array.hsrc/TiledArray/device/device_array_ops.h renamed to src/TiledArray/device/device_array.h

@@ -21,21 +21,24 @@
  *
  */
 
-#ifndef TILEDARRAY_DEVICE_ARRAY_OPERATIONS_H
-#define TILEDARRAY_DEVICE_ARRAY_OPERATIONS_H
+#ifndef TILEDARRAY_DEVICE_ARRAY_H
+#define TILEDARRAY_DEVICE_ARRAY_H
 
 #include <TiledArray/config.h>
 
 #ifdef TILEDARRAY_HAS_DEVICE
+#include <TiledArray/fwd.h>
 
 #include <TiledArray/device/um_storage.h>
 #include <TiledArray/external/device.h>
-#include <TiledArray/fwd.h>
 #include <TiledArray/tile.h>
 
 namespace TiledArray {
 
-/// Array-level to_device operation for DistArrays containing device tensors
+/// @brief Array-level to_device operation for DistArrays
+/// @tparam UMT Device (UM) Tile type
+/// @tparam Policy Policy for DistArray
+/// @param um_array input array
 template <typename UMT, typename Policy>
 void to_device(TiledArray::DistArray<TiledArray::Tile<UMT>, Policy> &um_array) {
   auto to_device_fn = [](TiledArray::Tile<UMT> &tile) {
@@ -66,7 +69,10 @@ void to_device(TiledArray::DistArray<TiledArray::Tile<UMT>, Policy> &um_array) {
   DeviceSafeCall(device::deviceSynchronize());
 }
 
-/// Array-level to_host operation for DistArrays containing device tensors
+/// @brief Array-level to_host operation for DistArrays
+/// @tparam UMT Device (UM) Tile type
+/// @tparam Policy Policy for DistArray
+/// @param um_array input array
 template <typename UMT, typename Policy>
 void to_host(TiledArray::DistArray<TiledArray::Tile<UMT>, Policy> &um_array) {
   auto to_host_fn = [](TiledArray::Tile<UMT> &tile) {
@@ -101,4 +107,4 @@ void to_host(TiledArray::DistArray<TiledArray::Tile<UMT>, Policy> &um_array) {
 
 #endif  // TILEDARRAY_HAS_DEVICE
 
-#endif  // TILEDARRAY_DEVICE_ARRAY_OPERATIONS_H
+#endif  // TILEDARRAY_DEVICE_ARRAY_H

src/TiledArray/device/um_tensor.h

@@ -31,7 +31,7 @@
 #include <TiledArray/fwd.h>
 
 #include <TiledArray/device/blas.h>
-#include <TiledArray/device/device_array_ops.h>
+#include <TiledArray/device/device_array.h>
 #include <TiledArray/device/kernel/mult_kernel.h>
 #include <TiledArray/device/kernel/reduce_kernel.h>
 #include <TiledArray/device/um_storage.h>