modify EV name (#376)

NVIDIA-Merlin · Feb 9, 2023 · 2cf6853 · 2cf6853
1 parent 44355bd
commit 2cf6853
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sparse_operation_kit/experiment/lookup/kernels/embedding_collection.cc b/sparse_operation_kit/experiment/lookup/kernels/embedding_collection.cc
@@ -920,7 +920,7 @@ namespace tensorflow {
 template <typename KeyType, typename OffsetType, typename DType>
 class LookupForwardEmbeddingVarGPUOp : public EmbeddingCollectionBase<KeyType, OffsetType, DType> {
  private:
-  using VarType = EmbeddingVarGPU<KeyType, float>;
+  using VarType = EmbeddingVar<KeyType, float>;
   EmbeddingVarGPUAdapter<KeyType, float> adapter_;
 
  public: