Searched refs:MASK_BYTES (Results 1 – 1 of 1) sorted by relevance
27 #define MASK_BYTES (MASK_WORDS * sizeof(int64)) macro44 se::DeviceMemory<int64> gpu_ptr{se::DeviceMemoryBase{ptr, MASK_BYTES}}; in CheckMask()47 Status result = exec->SynchronousMemcpyD2H(gpu_ptr, MASK_BYTES, tmp); in CheckMask()66 se::DeviceMemory<int64> gpu_ptr{se::DeviceMemoryBase{ptr, MASK_BYTES}}; in InitMask()67 Status result = exec->SynchronousMemcpyH2D(mask, MASK_BYTES, &gpu_ptr); in InitMask()89 num_bytes += (2 * MASK_BYTES); in AllocateRaw()94 void* rv = static_cast<char*>(allocated_ptr) + MASK_BYTES; in AllocateRaw()102 static_cast<char*>(allocated_ptr) + req_size - MASK_BYTES, in AllocateRaw()112 ptr = static_cast<void*>(static_cast<char*>(ptr) - MASK_BYTES); in DeallocateRaw()122 MASK_BYTES); in RequestedSize()[all …]