Searched refs:src_dev_ctx (Results 1 – 2 of 2) sorted by relevance
84 DeviceContext* src_dev_ctx, DeviceContext* dst_dev_ctx, Device* src_dev, in MemCpyAsync() argument103 if (src_dev_ctx == nullptr && src_device_type == DEVICE_GPU) { in MemCpyAsync()107 src_dev_ctx = dev_info->default_context; in MemCpyAsync()115 if (non_cpu_src) CHECK(src_dev_ctx); in MemCpyAsync()119 src_dev_ctx, dst_dev_ctx, src_dev, dst_dev, src_attr, in MemCpyAsync()
76 static void MemCpyAsync(DeviceContext* src_dev_ctx,