We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 73bcba9 commit 595c613Copy full SHA for 595c613
src/TiledArray/fwd.h
@@ -142,6 +142,10 @@ template <typename T, typename Range = TiledArray::Range>
142
using btasUMTensorVarray =
143
::btas::Tensor<T, Range, TiledArray::device_um_btas_varray<T>>;
144
145
+/// TA::Tensor with UM storage
146
+template <typename T>
147
+using UMTensorType = TiledArray::Tensor<T, TiledArray::device_um_allocator<T>>;
148
+
149
#endif // TILEDARRAY_HAS_DEVICE
150
151
template <typename>
0 commit comments