From 40525486620501665a7f515d4abf825becab618f Mon Sep 17 00:00:00 2001 From: Xinya Zhang Date: Wed, 17 Jul 2024 23:19:56 -0500 Subject: [PATCH] Fix the compiling error. --- .../training/backward/embedding_backward_split_template.cu | 1 + 1 file changed, 1 insertion(+) diff --git a/fbgemm_gpu/codegen/training/backward/embedding_backward_split_template.cu b/fbgemm_gpu/codegen/training/backward/embedding_backward_split_template.cu index 91829206d1..d143dd4c93 100644 --- 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 using Tensor = at::Tensor; using namespace fbgemm_gpu;