Coverage for gpaw/gpu/cpupy/cublas.py: 50%

1def syrk(trans, a, c, alpha, beta, lower): 

2 from gpaw.utilities.blas import rk 

3 

4 assert trans == 'N' 

5 assert lower 

6 rk(alpha, a._data, beta, c._data) 

7 

8 

9def gemm(transa, transb, a, b, c, alpha, beta): 

10 from gpaw.utilities.blas import mmm 

11 

12 mmm(alpha, 

13 a._data, 

14 transa.replace('H', 'C'), 

15 b._data, 

16 transb.replace('H', 'C'), 

17 beta, 

18 c._data)