Skip to content

Commit f106b02

Browse files
committed
UMTensor: implement is_device_tile
1 parent 1f400bb commit f106b02

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

src/TiledArray/device/um_tensor.h

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,12 @@
4747
namespace TiledArray {
4848
namespace detail {
4949

50+
/// is_device_tile specialization for UMTensor
51+
template <typename T>
52+
struct is_device_tile<
53+
::TiledArray::Tensor<T, TiledArray::device_um_allocator<T>>>
54+
: public std::true_type {};
55+
5056
/// pre-fetch to device
5157
template <typename T>
5258
void to_device(const UMTensor<T> &tensor) {

0 commit comments

Comments
 (0)