Searched refs:n_comp (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/tao/unconstrained/tutorials/ |
| H A D | rosenbrock4.h | 362 PetscInt n_comp = r.c_end - r.c_start; in RosenbrockObjective_Device() local
365 …if (n_comp) PetscCUPMLaunch(RosenbrockObjective_Kernel<<<(n_comp + 255) / 256, 256, 0, stream>>>(r… in RosenbrockObjective_Device()
366 PetscCall(PetscLogGpuFlops(RosenbrockObjectiveFlops * n_comp)); in RosenbrockObjective_Device()
372 PetscInt n_comp = r.c_end - r.c_start; in RosenbrockGradient_Device() local
375 …if (n_comp) PetscCUPMLaunch(RosenbrockGradient_Kernel<<<(n_comp + 255) / 256, 256, 0, stream>>>(r,… in RosenbrockGradient_Device()
376 PetscCall(PetscLogGpuFlops(RosenbrockGradientFlops * n_comp)); in RosenbrockGradient_Device()
382 PetscInt n_comp = r.c_end - r.c_start; in RosenbrockObjectiveGradient_Device() local
385 …if (n_comp) PetscCUPMLaunch(RosenbrockObjectiveGradient_Kernel<<<(n_comp + 255) / 256, 256, 0, str… in RosenbrockObjectiveGradient_Device()
386 PetscCall(PetscLogGpuFlops(RosenbrockObjectiveGradientFlops * n_comp)); in RosenbrockObjectiveGradient_Device()
392 PetscInt n_comp = r.c_end - r.c_start; in RosenbrockHessian_Device() local
[all …]
|