tensorflow/stream_executor/gpu/redzone_allocator.cc
// Run the redzone checker on the provided buffer redzone.
//
// Increment out_param if mismatch occurs.
Run the redzone checker on the provided buffer redzone.
Increment out_param if mismatch occurs.
Run the redzone checker on the provided buffer redzone . Increment out_param if mismatch occurs .
static void RunRedzoneChecker(Stream* stream,
                              const DeviceMemory<uint8>& redzone,
                              uint8 redzone_pattern,
                              const DeviceMemory<uint64_t>& out_param,
                              const ComparisonKernelT& comparison_kernel) {
  StreamExecutor* executor = stream->parent();
  int64_t num_elements = redzone.size();
  int64_t threads_per_block = std::min(
      executor->GetDeviceDescription().threads_per_block_limit(), num_elements);
  int64_t block_count =
      tensorflow::MathUtil::CeilOfRatio(num_elements, threads_per_block);
  stream->ThenLaunch(ThreadDim(threads_per_block), BlockDim(block_count),
                     comparison_kernel, redzone, redzone_pattern,
                     redzone.size(), out_param);
}
static void RunRedzoneChecker ( Stream * stream , const DeviceMemory < uint8 > & redzone , uint8 redzone_pattern , const DeviceMemory < uint64_t > & out_param , const ComparisonKernelT & comparison_kernel ) { StreamExecutor * executor = stream -> parent ( ) ; int64_t num_elements = redzone . size ( ) ; int64_t threads_per_block = std :: min ( executor -> GetDeviceDescription ( ) . threads_per_block_limit ( ) , num_elements ) ; int64_t block_count = tensorflow :: MathUtil :: CeilOfRatio ( num_elements , threads_per_block ) ; stream -> ThenLaunch ( ThreadDim ( threads_per_block ) , BlockDim ( block_count ) , comparison_kernel , redzone , redzone_pattern , redzone . size ( ) , out_param ) ; }
Run the redzone checker on the provided buffer redzone.
Run the redzone checker on the provided buffer redzone .