Fix MSVC+CUDA issues.

This commit is contained in:
Antonio Sánchez
2022-04-08 18:05:32 +00:00
parent 734ed1efa6
commit 5ed7a86ae9
8 changed files with 26 additions and 32 deletions

View File

@@ -244,7 +244,7 @@ class TensorBlockDescriptor {
const DestinationBufferKind& kind() const { return m_kind; }
private:
friend class TensorBlockDescriptor;
friend class TensorBlockDescriptor<NumDims, IndexType>;
DestinationBuffer() : m_data(NULL), m_data_type_size(0), m_kind(kEmpty) {}
@@ -708,7 +708,7 @@ class TensorMaterializedBlock {
}
private:
friend class TensorMaterializedBlock;
friend class TensorMaterializedBlock<Scalar, NumDims, Layout, IndexType>;
Storage(Scalar* data, const Dimensions& dimensions,
const Dimensions& strides, bool materialized_in_output,

View File

@@ -235,7 +235,7 @@ EigenContractionKernelInternal(const LhsMapper lhs, const RhsMapper rhs,
} \
} \
#define writeRegToShmem(_) \
#define writeRegToShmem() \
lhs_shmem[lhs_store_idx_0] = lhs_pf0; \
rhs_shmem[rhs_store_idx_0] = rhs_pf0; \
\