remove unused code

AlibabaPAI · Sep 11, 2024 · 48d0786 · 48d0786
1 parent a599133
commit 48d0786
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/torch_xla/csrc/runtime/torch_allocator.cc b/torch_xla/csrc/runtime/torch_allocator.cc
@@ -38,10 +38,6 @@ void TorchCUDACachingAllocator::DeallocateRaw(void* ptr) {
 
 void TorchCUDACachingAllocator::SetStreamAndPreallocateMemory(void* stream) {
   auto new_cuda_stream = static_cast<cudaStream_t>(stream);
-  if (cuda_stream_ != nullptr && new_cuda_stream != cuda_stream_) {
-    LOG(FATAL) <<  // Crash OK.
-        "Trying to set the stream twice. This isn't supported. ";
-  }
   VLOG(3) << "Setting cuda stream " << stream
           << " for TorchCUDACachingAllocator on device "
           << static_cast<int>(device_index_);