Function checkAndMigrate

src/backend/cuda/Array.cpp:61–74

template<typename T>
void checkAndMigrate(Array<T> &arr) {
    int arr_id = arr.getDevId();
    int cur_id = detail::getActiveDeviceId();
    if (!isDeviceBufferAccessible(arr_id, cur_id)) {
        static auto getLogger = [&] { return spdlog::get("platform"); };
        AF_TRACE("Migrating array from {} to {}.", arr_id, cur_id);
        auto migrated_data = memAlloc<T>(arr.elements());
        CUDA_CHECK(
            cudaMemcpyPeerAsync(migrated_data.get(), getDeviceNativeId(cur_id),
                                arr.get(), getDeviceNativeId(arr_id),
                                arr.elements() * sizeof(T), getActiveStream()));
        arr.data.reset(migrated_data.release(), memFree);
    }
}

template<typename T>
Array<T>::Array(const af::dim4 &dims)