experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[libtpu] Fix static_asserts.

Browse Source 
PiperOrigin-RevId: 337992564
Change-Id: Ieba57b091ee23923eb1a6701bd80725fd891c80a
This commit is contained in:
Skye Wanderman-Milne 2020-10-19 21:22:50 -07:00 committed by TensorFlower Gardener
parent f41a9a9335
commit 643bcb35e6
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/stream_executor/tpu/c_api_conversions.cc
View File

@ -156,7 +156,7 @@ stream_executor::DeviceMemoryBase FromC(const SE_DeviceMemoryBase& se_base) {
// types that require a static_cast.
template <typename Src, typename Dst, typename DstList>
static void CopyVectorBase(const absl::Span<Src> src, DstList* dst) {
static_assert(sizeof(Src) == sizeof(Dst));
static_assert(sizeof(Src) == sizeof(Dst), "Mismatched types");
dst->size = src.size();
if (dst->size > TPU_C_API_MAX_INLINED) {
dst->heap = new Dst[dst->size];
@ -196,7 +196,7 @@ static void CopyVector(const absl::Span<const xla::Tile> src, TileList* dst) {
// types that require a static_cast.
template <typename Dst, typename Src, typename SrcList>
static absl::Span<const Dst> MakeSpanBase(const SrcList& src_list) {
static_assert(sizeof(Src) == sizeof(Dst));
static_assert(sizeof(Src) == sizeof(Dst), "Mismatched types");
const Src* src = src_list.size > TPU_C_API_MAX_INLINED ? src_list.heap
: &src_list.inlined[0];
return absl::Span<const Dst>(reinterpret_cast<const Dst*>(src),