Searched refs:dev_ptr (Results 1 – 6 of 6) sorted by relevance
83 thrust::device_ptr<T> dev_ptr(array); in ThrustFillWith() local84 thrust::fill(thrust::cuda::par.on(cuda_stream), dev_ptr, dev_ptr + size, tofill); in ThrustFillWith()
82 void GPUBucket::FreeDeviceMem(void *dev_ptr) { GPUMemoryAllocator::GetInstance().FreeTensorMem(dev_… in FreeDeviceMem() argument
35 void FreeDeviceMem(void *dev_ptr) override;
75 void AscendBucket::FreeDeviceMem(void *dev_ptr) { AscendMemoryPool::GetInstance().FreeTensorMem(dev… in FreeDeviceMem() argument
85 virtual void FreeDeviceMem(void *dev_ptr) = 0;