Skip to content

Commit

Permalink
More GPU macro protection
Browse files Browse the repository at this point in the history
  • Loading branch information
DylanMcKinney authored Jul 30, 2018
1 parent 783dd24 commit 07f4897
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 07f4897

Please sign in to comment.