fix last indent bug

cbd86de8 · Rick Ho · 2ed4351d · cbd86de8
Commit cbd86de8 authored Jan 09, 2021 by Rick Ho
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 3 deletions

pytorch/cuda/moe_cuda_kernel.cu pytorch/cuda/moe_cuda_kernel.cu +4 -3

No files found.
--- a/pytorch/cuda/moe_cuda_kernel.cu
+++ b/pytorch/cuda/moe_cuda_kernel.cu
@@ -7,7 +7,7 @@

 #include <cuda.h>
 #include <cuda_runtime.h>
-#include <cublas_v2.h>                                                                                          
+#include <cublas_v2.h>
 #include <helper_cuda.h> 

 #include <mpi.h>
@@ -143,8 +143,9 @@ void moe_cuda_forward_impl(

 #ifdef MOE_BREAKDOWN
 	timestamp(t_expert);
-	fprintf(stderr, "Expert asn time %.3lf us\n", getDuration(t_init, t_expert) *
-			1e6);
+	fprintf(stderr, "Expert asn %d time %.3lf us\n", 
+			expert_sz,
+			getDuration(t_init, t_expert) * 1e6);
 #endif

 	batch_scatter_kernel<scalar_t>