Searched refs:spgemm_alg (Results 1 – 2 of 2) sorted by relevance
1068 …auto spgemm_alg = KokkosSparse::SPGEMMAlgorithm::SPGEMM_DEFAULT; // default is TPL if enabled, oth… in MatProductSymbolic_MPIAIJKokkos_AtB() local1073 spgemm_alg = KokkosSparse::SPGEMMAlgorithm::SPGEMM_KK; in MatProductSymbolic_MPIAIJKokkos_AtB()1077 PetscCallCXX(mm->kh1.create_spgemm_handle(spgemm_alg)); in MatProductSymbolic_MPIAIJKokkos_AtB()1078 PetscCallCXX(mm->kh2.create_spgemm_handle(spgemm_alg)); in MatProductSymbolic_MPIAIJKokkos_AtB()1079 PetscCallCXX(mm->kh3.create_spgemm_handle(spgemm_alg)); in MatProductSymbolic_MPIAIJKokkos_AtB()1080 PetscCallCXX(mm->kh4.create_spgemm_handle(spgemm_alg)); in MatProductSymbolic_MPIAIJKokkos_AtB()1182 …auto spgemm_alg = KokkosSparse::SPGEMMAlgorithm::SPGEMM_DEFAULT; // default is TPL if enabled, oth… in MatProductSymbolic_MPIAIJKokkos_AB() local1187 spgemm_alg = KokkosSparse::SPGEMMAlgorithm::SPGEMM_KK; in MatProductSymbolic_MPIAIJKokkos_AB()1191 mm->kh1.create_spgemm_handle(spgemm_alg); in MatProductSymbolic_MPIAIJKokkos_AB()1192 mm->kh2.create_spgemm_handle(spgemm_alg); in MatProductSymbolic_MPIAIJKokkos_AB()[all …]
946 …auto spgemm_alg = KokkosSparse::SPGEMMAlgorithm::SPGEMM_DEFAULT; /* default alg is TPL if enabled,… in MatProductSymbolic_SeqAIJKokkos_SeqAIJKokkos() local951 spgemm_alg = KokkosSparse::SPGEMMAlgorithm::SPGEMM_KK; in MatProductSymbolic_SeqAIJKokkos_SeqAIJKokkos()954 PetscCallCXX(pdata->kh.create_spgemm_handle(spgemm_alg)); in MatProductSymbolic_SeqAIJKokkos_SeqAIJKokkos()