aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/stream_executor/cuda/cuda_gpu_executor.h
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/stream_executor/cuda/cuda_gpu_executor.h')
-rw-r--r--tensorflow/stream_executor/cuda/cuda_gpu_executor.h10
1 files changed, 5 insertions, 5 deletions
diff --git a/tensorflow/stream_executor/cuda/cuda_gpu_executor.h b/tensorflow/stream_executor/cuda/cuda_gpu_executor.h
index 1481dcc19a..53b2a29ae7 100644
--- a/tensorflow/stream_executor/cuda/cuda_gpu_executor.h
+++ b/tensorflow/stream_executor/cuda/cuda_gpu_executor.h
@@ -70,16 +70,16 @@ class CUDAExecutor : public internal::StreamExecutorInterface {
const BlockDim &block_dims, const KernelBase &k,
const KernelArgsArrayBase &args) override;
- int CalculateOccupancy(const DeviceDescription& device_description,
+ int CalculateOccupancy(const DeviceDescription &device_description,
uint64 registers_per_thread,
uint64 shared_memory_per_block,
- const ThreadDim& thread_dims, CUfunction func);
+ const ThreadDim &thread_dims, CUfunction func);
- int CompareOccupancy(int* initial_blocks,
- const DeviceDescription& device_description,
+ int CompareOccupancy(int *initial_blocks,
+ const DeviceDescription &device_description,
uint64 registers_per_thread,
uint64 shared_memory_per_block,
- const ThreadDim& thread_dims, CUfunction func);
+ const ThreadDim &thread_dims, CUfunction func);
void *Allocate(uint64 size) override;