Skip to content

Commit

Permalink
Merge pull request elemental#34 from DylanMcKinney/patch-2
Browse files Browse the repository at this point in the history
More GPU macro protection
  • Loading branch information
benson31 authored Jul 31, 2018
2 parents 783dd24 + 07f4897 commit 9f83369
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion include/El/blas_like/level1/Copy/internal_decl.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -64,11 +64,13 @@ template<typename T,Dist U,Dist V,
void TransposeDist( DistMatrix<T,U,V,ELEMENT,Device::CPU> const& A,
DistMatrix<T,V,U,ELEMENT,Device::CPU>& B );

#ifdef HYDROGEN_HAVE_CUDA
template<typename T,Dist U,Dist V,
typename=EnableIf<IsDeviceValidType<T,Device::GPU>>>
void TransposeDist( DistMatrix<T,U,V,ELEMENT,Device::GPU> const& A,
DistMatrix<T,V,U,ELEMENT,Device::GPU>& B );

#endif

template<typename T,Dist U,Dist V,Device D,
typename=DisableIf<IsDeviceValidType<T,D>>,typename=void>
void TransposeDist( DistMatrix<T,U,V,ELEMENT,D> const& A,
Expand Down

0 comments on commit 9f83369

Please sign in to comment.