pub unsafe fn tracegen( d_count: &DeviceBuffer<F>, d_cpu_count: &Option<DeviceBuffer<u32>>, d_trace: &DeviceBuffer<F>, d_sizes: &DeviceBuffer<u32>, stream: cudaStream_t, ) -> Result<(), CudaError>