diff --git a/fbgemm_gpu/src/dram_kv_embedding_cache/dram_kv_embedding_inference_wrapper.cpp b/fbgemm_gpu/src/dram_kv_embedding_cache/dram_kv_embedding_inference_wrapper.cpp index e69b66aa00..3e0afce46a 100644 --- a/fbgemm_gpu/src/dram_kv_embedding_cache/dram_kv_embedding_inference_wrapper.cpp +++ b/fbgemm_gpu/src/dram_kv_embedding_cache/dram_kv_embedding_inference_wrapper.cpp @@ -35,6 +35,11 @@ void DramKVEmbeddingInferenceWrapper::init( static_cast(std::get<2>(specs[0])), static_cast(row_alignment), static_cast(scale_bias_size_in_bytes)); + LOG(INFO) << "Initialize dram_kv with max_D: " << max_D + << ", sparse_type: " << std::get<2>(specs[0]) + << ", row_alignment: " << row_alignment + << ", scale_bias_size_in_bytes: " << scale_bias_size_in_bytes + << ", max_row_bytes_: " << max_row_bytes_; if (dram_kv_ != nullptr) { return; }