Skip to content

Commit 595c613

Browse files
committed
fwd: try introducing a TA::Tensor with UM allocator [skip ci]
1 parent 73bcba9 commit 595c613

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/TiledArray/fwd.h

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -142,6 +142,10 @@ template <typename T, typename Range = TiledArray::Range>
142142
using btasUMTensorVarray =
143143
::btas::Tensor<T, Range, TiledArray::device_um_btas_varray<T>>;
144144

145+
/// TA::Tensor with UM storage
146+
template <typename T>
147+
using UMTensorType = TiledArray::Tensor<T, TiledArray::device_um_allocator<T>>;
148+
145149
#endif // TILEDARRAY_HAS_DEVICE
146150

147151
template <typename>

0 commit comments

Comments
 (0)