message: [SE] Enforce a minimum allocation alignment of 64 bytes for the Host DeviceMemoryAllocator.
  Will fix when incorporated into a jaxlib release.
diff: @@ -20,6 +20,7 @@ limitations under the License.
  #include <string.h>
  #include "absl/synchronization/notification.h"
  +#include "tensorflow/core/platform/mem.h"
  #include "tensorflow/core/platform/profile_utils/cpu_utils.h"
  #include "tensorflow/stream_executor/host/host_platform_id.h"
  #include "tensorflow/stream_executor/host/host_stream.h"
  @@ -43,7 +44,11 @@ HostExecutor::~HostExecutor() {}
  DeviceMemoryBase HostExecutor::Allocate(uint64 size, int64 memory_space) {
  CHECK_EQ(memory_space, 0);
  - return DeviceMemoryBase(new char[size], size);
  + // Use a minimum alignment of 64 bytes to be friendly to AVX512 code.
  + // This should probably be kept in sync with
  + // tensorflow::Allocator::kAllocatorAlignment.
  + return DeviceMemoryBase(
  + tensorflow::port::AlignedMalloc(size, /*minimum_alignment=*/64), size);
  }
  void *HostExecutor::GetSubBuffer(DeviceMemoryBase *parent, uint64 offset_bytes,
  @@ -52,7 +57,7 @@ void *HostExecutor::GetSubBuffer(DeviceMemoryBase *parent, uint64 offset_bytes,
  }
  void HostExecutor::Deallocate(DeviceMemoryBase *mem) {
  - delete[] static_cast<char *>(mem->opaque());
  + tensorflow::port::AlignedFree(mem->opaque());
  }
  port::Status HostExecutor::SynchronousMemZero(DeviceMemoryBase *location,
  