update

pytorch/cuda/cuda_stream_manager.h

@@ -16,11 +16,13 @@ public:
         which means pytorch always sets the correct device for us.
         But for safety, we still manually set device to the desired one.
         */
-
+        /*
         int current_device;
         checkCudaErrors(cudaGetDevice(&current_device));
         printf("CudaStreamManager construnctor called, get device %d, set device %d\n", current_device, device);
+        */
         checkCudaErrors(cudaSetDevice(device));
+        
         streams = new cudaStream_t[num_expert];
         checkCudaErrors(cublasCreate(&handle));
         for (size_t i=0; i<num_expert; ++i) {