Fix the compiling error.

ROCm · Aug 7, 2024 · 4052548 · 4052548
1 parent 060c759
commit 4052548
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/fbgemm_gpu/codegen/training/backward/embedding_backward_split_template.cu b/fbgemm_gpu/codegen/training/backward/embedding_backward_split_template.cu
@@ -22,6 +22,7 @@
 #include "fbgemm_gpu/split_embeddings_utils.cuh"
 #include "fbgemm_gpu/sparse_ops.h"
 #include "fbgemm_gpu/sparse_ops_utils.h"
+#include <iostream>
 
 using Tensor = at::Tensor;
 using namespace fbgemm_gpu;