Skip to content

GitLab

Commit cbd86de8 authored by Rick Ho's avatar Rick Ho
Browse files

fix last indent bug

parent 2ed4351d
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 3 deletions
pytorch/cuda/moe_cuda_kernel.cu
View file @ cbd86de8
...@@ -7,7 +7,7 @@ ...@@ -7,7 +7,7 @@
#include <cuda.h> #include <cuda.h>
#include <cuda_runtime.h> #include <cuda_runtime.h>
#include <cublas_v2.h> #include <cublas_v2.h>
#include <helper_cuda.h> #include <helper_cuda.h>
#include <mpi.h> #include <mpi.h>
...@@ -143,8 +143,9 @@ void moe_cuda_forward_impl( ...@@ -143,8 +143,9 @@ void moe_cuda_forward_impl(
#ifdef MOE_BREAKDOWN #ifdef MOE_BREAKDOWN
timestamp(t_expert); timestamp(t_expert);
fprintf(stderr, "Expert asn time %.3lf us\n", getDuration(t_init, t_expert) * fprintf(stderr, "Expert asn %d time %.3lf us\n",
1e6); expert_sz,
getDuration(t_init, t_expert) * 1e6);
#endif #endif
batch_scatter_kernel<scalar_t> batch_scatter_kernel<scalar_t>
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment