@@ -235,8 +235,8 @@ PBAT_HOST_DEVICE auto Unit(auto i)
235
235
return Identity<TScalar, M, M>().Col (i);
236
236
}
237
237
238
- template <int M, int N, class TScalar , class IndexType = std:: int64_t >
239
- PBAT_HOST_DEVICE auto FromFlatBuffer (TScalar* buf, IndexType bi)
238
+ template <int M, int N, class TScalar >
239
+ PBAT_HOST_DEVICE auto FromFlatBuffer (TScalar* buf, std:: int64_t bi)
240
240
{
241
241
return SMatrixView<TScalar, M, N>(buf + M * N * bi);
242
242
}
@@ -255,9 +255,9 @@ PBAT_HOST_DEVICE auto FromFlatBuffer(TScalar* buf, TIndexMatrix const& inds)
255
255
return A;
256
256
}
257
257
258
- template <CMatrix TMatrix, class IndexType = std:: int64_t >
258
+ template <CMatrix TMatrix>
259
259
PBAT_HOST_DEVICE void
260
- ToFlatBuffer (TMatrix const & A, typename TMatrix::ScalarType* buf, IndexType bi)
260
+ ToFlatBuffer (TMatrix const & A, typename TMatrix::ScalarType* buf, std:: int64_t bi)
261
261
{
262
262
auto constexpr M = TMatrix::kRows ;
263
263
auto constexpr N = TMatrix::kCols ;
@@ -290,8 +290,8 @@ ToFlatBuffer(TMatrix const& A, TIndexMatrix const& inds, typename TMatrix::Scala
290
290
}
291
291
}
292
292
293
- template <int M, int N, class TScalar , class IndexType = std:: int64_t >
294
- PBAT_HOST_DEVICE auto FromBuffers (std::array<TScalar*, M> buf, IndexType bi)
293
+ template <int M, int N, class TScalar >
294
+ PBAT_HOST_DEVICE auto FromBuffers (std::array<TScalar*, M> buf, std:: int64_t bi)
295
295
{
296
296
using ScalarType = std::remove_const_t <TScalar>;
297
297
SMatrix<ScalarType, M, N> A{};
@@ -315,9 +315,9 @@ PBAT_HOST_DEVICE auto FromBuffers(std::array<TScalar*, K> buf, TIndexMatrix cons
315
315
return A;
316
316
}
317
317
318
- template <CMatrix TMatrix, int M, class IndexType = std:: int64_t >
318
+ template <CMatrix TMatrix, int M>
319
319
PBAT_HOST_DEVICE void
320
- ToBuffers (TMatrix const & A, std::array<typename TMatrix::ScalarType*, M> buf, IndexType bi)
320
+ ToBuffers (TMatrix const & A, std::array<typename TMatrix::ScalarType*, M> buf, std:: int64_t bi)
321
321
{
322
322
static_assert (M == TMatrix::kRows , " A must have same rows as number of buffers" );
323
323
auto constexpr N = TMatrix::kCols ;
0 commit comments