• Frederic's avatar
    finish gpu ger tests: · c500bafa
    Frederic 提交于
    -renamed TestGer_local_gemm_to_ger to TestGer
    -changed the Ger.infer_shape methode with ger.perform
     This make test of cpu and gpu ger op easier and follow what is done most of the time
    -Moved tests in TestGer_make_thunk to TestGer and use theano.function
     This was needed to make them work on cpu and gpu
    -make TestGer always call call the compiled function
    c500bafa
blas.py 61.9 KB