| 69 | } |
| 70 | |
| 71 | THolder<TCholeskySolverKernel::TKernelContext> TCholeskySolverKernel::PrepareContext(IMemoryManager& manager) const { |
| 72 | const ui32 rowSize = Solutions.ObjectSize(); |
| 73 | if (!TKernelContext::UseCuSolver(rowSize, Matrices.ObjectCount())) { |
| 74 | return MakeHolder<TKernelContext>(); |
| 75 | } |
| 76 | |
| 77 | auto context = MakeHolder<TKernelContext>(rowSize); |
| 78 | context->AllocateBuffers(manager); |
| 79 | |
| 80 | return context; |
| 81 | } |
| 82 | |
| 83 | void TCholeskySolverKernel::Run(const TCudaStream& stream, TCholeskySolverKernel::TKernelContext& context) const { |
| 84 | const ui32 rowSize = Solutions.ObjectSize(); |
nothing calls this directly
no test coverage detected