Home
last modified time | relevance | path

Searched refs:gemmBatched (Results 1 – 15 of 15) sorted by relevance

/dports/science/qmcpack/qmcpack-3.11.0/src/AFQMC/HamiltonianOperations/
H A DReal3IndexFactorization_batched_v2.hpp709 using ma::gemmBatched; in generalizedFockMatrix()
713gemmBatched('T', 'N', NMO, NMO, NMO * local_nCV, SPComplexType(1.0), Aarray.data(), NMO * local_nC… in generalizedFockMatrix()
726 using ma::gemmBatched; in generalizedFockMatrix()
729gemmBatched('T', 'N', NMO * NMO, 1, local_nCV, SPComplexType(-1.0) * scl, Aarray.data(), local_nCV, in generalizedFockMatrix()
748 using ma::gemmBatched; in generalizedFockMatrix()
751gemmBatched('T', 'N', NMO * NMO, 1, local_nCV, SPComplexType(-1.0) * scl, Aarray.data(), local_nCV, in generalizedFockMatrix()
771 using ma::gemmBatched; in generalizedFockMatrix()
773gemmBatched('T', 'N', NMO, NMO, NMO * local_nCV, SPComplexType(1.0), Aarray.data(), NMO * local_nC… in generalizedFockMatrix()
809 using ma::gemmBatched; in generalizedFockMatrix()
811gemmBatched('T', 'N', NMO, NMO, NMO, ComplexType(-1.0), Aarr.data(), NMO, Barr.data(), NMO, Comple… in generalizedFockMatrix()
[all …]
H A DKP3IndexFactorization_batched.hpp613 using ma::gemmBatched; in getOneBodyPropagatorMatrix()
715gemmBatched('T', 'N', nocc_max * nchol_max, nwalk * nocc_max, npol * nmo_max, SPComplexType(1.0), in getOneBodyPropagatorMatrix()
746gemmBatched('T', 'N', nocc_max * nchol_max, nwalk * nocc_max, npol * nmo_max, SPComplexType(1.0), in getOneBodyPropagatorMatrix()
1258 using ma::gemmBatched; in getOneBodyPropagatorMatrix()
1282gemmBatched('T', 'T', nmo_max2, nwalk, nchol_max, SPComplexType(1.0), Aarray.data(), nchol_max, Ba… in getOneBodyPropagatorMatrix()
1316gemmBatched('C', 'T', nmo_max2, nwalk, nchol_max, SPComplexType(1.0), Aarray.data(), nchol_max, Ba… in getOneBodyPropagatorMatrix()
1370 using ma::gemmBatched; in getOneBodyPropagatorMatrix()
1452gemmBatched('N', 'T', nwalk, nchol_max, Kak, SPComplexType(1.0), Aarray.data(), nwalk, Barray.data… in getOneBodyPropagatorMatrix()
/dports/science/qmcpack/qmcpack-3.11.0/src/AFQMC/SlaterDeterminantOperations/
H A Dmixed_density_matrix.hpp989 using ma::gemmBatched; in MixedDensityMatrix()
1071gemmBatched('T', 'T', NMO, NEL, NEL, ComplexType(1.0), Warray.data(), ldw, NNarray.data(), ldN, Co… in MixedDensityMatrix()
1090gemmBatched('T', 'T', NMO, NEL, NEL, ComplexType(1.0), Warray.data(), ldw, NNarray.data(), ldN, Co… in MixedDensityMatrix()
1117gemmBatched('T', 'T', NMO, NMO, NEL, ComplexType(1.0), Warray.data(), ldw, NMarray.data(), ldM, Co… in MixedDensityMatrix()
1145 using ma::gemmBatched; in DensityMatrices()
1200gemmBatched('N', 'N', NEL, NEL, NMO, ComplexType(1.0), Rarray.data(), ldR, Larray.data(), ldL, Com… in DensityMatrices()
1203gemmBatched('N', 'C', NEL, NEL, NMO, ComplexType(1.0), Rarray.data(), ldR, Larray.data(), ldL, Com… in DensityMatrices()
1221gemmBatched('T', 'T', NMO, NEL, NEL, ComplexType(1.0), Rarray.data(), ldR, NNarray.data(), ldN, Co… in DensityMatrices()
1235gemmBatched('T', 'T', NMO, NEL, NEL, ComplexType(1.0), Rarray.data(), ldR, NNarray.data(), ldN, Co… in DensityMatrices()
1239gemmBatched('N', 'T', NMO, NMO, NEL, ComplexType(1.0), NMarray.data(), ldM, Larray.data(), ldL, Co… in DensityMatrices()
[all …]
H A Dapply_expM.hpp263 using ma::gemmBatched; in apply_expM_noncollinear()
265gemmBatched('N', TA, M, N, K, fact, pT1i->data(), (*pT1).stride(1), Vi.data(), ldv, zero, pT2i->da… in apply_expM_noncollinear()
/dports/science/qmcpack/qmcpack-3.11.0/src/AFQMC/Numerics/detail/CUDA/
H A Dold_blas_cuda.hpp542 inline static void gemmBatched(char Atrans, in gemmBatched() function
593 inline static void gemmBatched(char Atrans, in gemmBatched() function
621 using ma::gemmBatched; in gemmBatched()
622 gemmBatched(Atrans, Btrans, M, N, K, alpha, A_d, lda, B_d, ldb, beta, C_d, ldc, batchSize); in gemmBatched()
H A Dblas_cuda_catch_all.hpp225 inline static void gemmBatched(char Atrans, in gemmBatched() function
H A Dblas_cuda_gpu_ptr.hpp314 inline static void gemmBatched(char Atrans, in gemmBatched() function
368 inline static void gemmBatched(char Atrans, in gemmBatched() function
/dports/science/qmcpack/qmcpack-3.11.0/src/AFQMC/Numerics/detail/HIP/
H A Dblas_hip_catch_all.hpp222 inline static void gemmBatched(char Atrans, in gemmBatched() function
H A Dblas_hip_gpu_ptr.hpp309 inline static void gemmBatched(char Atrans, in gemmBatched() function
363 inline static void gemmBatched(char Atrans, in gemmBatched() function
/dports/science/qmcpack/qmcpack-3.11.0/src/AFQMC/Estimators/Observables/
H A Datomcentered_correlators.hpp240 using ma::gemmBatched; in accumulate_reference()
335gemmBatched('N', 'N', NMO, int(iN - i0), NMO, ComplexType(1.0), Barray.data(), NMO, Aarray.data(),… in accumulate_reference()
349gemmBatched('T', 'N', NAO, int(iN - i0), NMO, ComplexType(1.0), Barray.data(), NMO, Carray.data(),… in accumulate_reference()
H A Drealspace_correlators.hpp225 using ma::gemmBatched; in accumulate_reference()
304gemmBatched('N', 'C', npts, npts, NMO, ComplexType(1.0), Barray.data(), npts, Aarray.data(), npts, in accumulate_reference()
/dports/science/qmcpack/qmcpack-3.11.0/src/AFQMC/Numerics/performance/
H A Dperformance.cpp201 using ma::gemmBatched; in timeBatchedGemm()
203gemmBatched('N', 'N', m, m, m, alpha, A_array.data(), m, B_array.data(), m, beta, C_array.data(), … in timeBatchedGemm()
/dports/science/qmcpack/qmcpack-3.11.0/src/AFQMC/Numerics/tests/
H A Dtest_dense_numerics.cpp656 using ma::gemmBatched; in test_dense_batched_gemm()
657gemmBatched('N', 'N', 3, 3, 3, alpha, A_array.data(), 3, B_array.data(), 3, beta, C_array.data(), … in test_dense_batched_gemm()
/dports/science/qmcpack/qmcpack-3.11.0/src/AFQMC/Numerics/
H A Dma_operations.hpp388 using ma::gemmBatched; in BatchedProduct()
389gemmBatched(TB, TA, M, N, K, element(alpha), Bi.data(), ldb, Ai.data(), lda, element(beta), Ci.dat… in BatchedProduct()
/dports/science/qmcpack/qmcpack-3.11.0/src/AFQMC/Numerics/detail/CPU/
H A Dblas_cpu.hpp1219 inline static void gemmBatched(char Atrans, in gemmBatched() function
1254 inline static void gemmBatched(char Atrans, in gemmBatched() function