Skip to content

Commit 238c0a8

Browse files
committed
UMTensor: add extern template declarations for UMTensor types
1 parent a816229 commit 238c0a8

File tree

1 file changed

+18
-0
lines changed

1 file changed

+18
-0
lines changed

src/TiledArray/device/um_tensor.h

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -768,6 +768,24 @@ struct ArchiveLoadImpl<Archive, TiledArray::UMTensor<T>> {
768768
} // namespace archive
769769
} // namespace madness
770770

771+
772+
#ifndef TILEDARRAY_HEADER_ONLY
773+
774+
namespace TiledArray {
775+
776+
extern template class Tensor<double, device_um_allocator<double>>;
777+
extern template class Tensor<float, device_um_allocator<float>>;
778+
extern template class Tensor<std::complex<double>,
779+
device_um_allocator<std::complex<double>>>;
780+
extern template class Tensor<std::complex<float>,
781+
device_um_allocator<std::complex<float>>>;
782+
extern template class Tensor<int, device_um_allocator<int>>;
783+
extern template class Tensor<long, device_um_allocator<long>>;
784+
785+
}
786+
787+
#endif // TILEDARRAY_HEADER_ONLY
788+
771789
#endif // TILEDARRAY_HAS_DEVICE
772790

773791
#endif // TILEDARRAY_DEVICE_UM_TENSOR_H

0 commit comments

Comments
 (0)