Fix MSVC+CUDA issues.

(cherry picked from commit 5ed7a86ae9)
This commit is contained in:
Antonio Sánchez
2022-04-08 18:05:32 +00:00
committed by Antonio Sanchez
parent 86d958e8f2
commit af912a7b5c
8 changed files with 27 additions and 33 deletions

View File

@@ -242,7 +242,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) {}
@@ -706,7 +706,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

@@ -233,7 +233,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; \
\