Searched refs:stashed_for_allocator_safety_ (Results 1 – 2 of 2) sorted by relevance
654 stashed_for_allocator_safety_->clear(); in synchronizeStream()2640 work->stashed_for_allocator_safety_ = in checkForNCCLErrorsInternal()2720 work->stashed_for_allocator_safety_ = in checkForNCCLErrorsInternal()2903 work->stashed_for_allocator_safety_ = in checkForNCCLErrorsInternal()3927 auto& v = work->stashed_for_allocator_safety_; in checkForNCCLErrorsInternal()4310 auto& v = work->stashed_for_allocator_safety_; in checkForNCCLErrorsInternal()
432 std::shared_ptr<std::vector<at::Tensor>> stashed_for_allocator_safety_; member in c10d::ProcessGroupNCCL::WorkNCCL