From 85a265031bbd61f5924cdbfdc316df6979883581 Mon Sep 17 00:00:00 2001
From: Benjamin Kramer <kramerb@google.com>
Date: Fri, 27 Jul 2018 11:04:02 -0700
Subject: [XLA:GPU] Only add the cubin if it is available

It's only non-empty if we were able to run ptxas. If the PTX is going to be
JIT'ed by the driver it won't be around. Loading an empty cubin will result in
a fatal error.

PiperOrigin-RevId: 206341931
---
 tensorflow/stream_executor/module_spec.h | 1 +
 1 file changed, 1 insertion(+)

(limited to 'tensorflow/stream_executor')

diff --git a/tensorflow/stream_executor/module_spec.h b/tensorflow/stream_executor/module_spec.h
index 212ae7ba9c..75bdfed2d7 100644
--- a/tensorflow/stream_executor/module_spec.h
+++ b/tensorflow/stream_executor/module_spec.h
@@ -43,6 +43,7 @@ class MultiModuleLoaderSpec {
   }
 
   void AddCudaCubinInMemory(port::ArraySlice<const uint8> cubin_bytes) {
+    CHECK(!cubin_bytes.empty());
     has_cuda_cubin_in_memory_ = true;
     cuda_cubin_in_memory_ = cubin_bytes;
   }
-- 
cgit v1.2.3