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

Fix the namespace for kCudaPlatformId, so cuda_platform.h is consistent with

Browse Source 
cuda_platform_id.h

PiperOrigin-RevId: 231504507
This commit is contained in:
Tim Shen 2019-01-29 17:17:08 -08:00 committed by TensorFlower Gardener
parent 91c36e137f
commit 8148037cde
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/stream_executor/cuda/cuda_platform.h
View File

@ -32,13 +32,14 @@ limitations under the License.
#include "tensorflow/stream_executor/trace_listener.h" #include "tensorflow/stream_executor/trace_listener.h"
namespace stream_executor { namespace stream_executor {
namespace gpu { namespace cuda {
// Opaque and unique identifier for the CUDA platform plugin. // Opaque and unique identifier for the CUDA platform plugin.
// This is needed so that plugins can refer to/identify this platform without // This is needed so that plugins can refer to/identify this platform without
// instantiating a CudaPlatform object. // instantiating a CudaPlatform object.
extern const Platform::Id kCudaPlatformId; extern const Platform::Id kCudaPlatformId;
} // namespace cuda
namespace gpu {
// Cuda-specific platform plugin, registered as a singleton value via module // Cuda-specific platform plugin, registered as a singleton value via module
// initializer. // initializer.
class CudaPlatform : public Platform { class CudaPlatform : public Platform {