Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
fix indent
Browse files Browse the repository at this point in the history
  • Loading branch information
arcadiaphy committed May 16, 2019
1 parent 10a5aa6 commit 8d16468
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion src/operator/linalg_impl.h
Original file line number Diff line number Diff line change
Expand Up @@ -1256,10 +1256,13 @@ LINALG_CPU_GETRF(dgetrf, double)

#ifdef __CUDACC__

// "getrfBatched" and "getriBatched" in cuBLAS must have DType *matrices[] as input
// to store the pointers of each batch matrix. This kernel is used to build the
// pointer array.
struct set_matrix : public mxnet::op::mxnet_op::tunable {
template<typename DType>
MSHADOW_XINLINE static void Map(int i, DType **p, DType *m, int step) {
p[i] = m + i * step;
p[i] = m + i * step;
}
};

Expand Down

0 comments on commit 8d16468

Please sign in to comment.