fix bugs in D76548519 (#4378)

Summary:
Pull Request resolved: #4378

X-link: facebookresearch/FBGEMM#1448

fix bugs that add the wrong check

Reviewed By: Raahul46

Differential Revision: D76944968

fbshipit-source-id: 0117b72afe72b3dca7d01c8511d437dfa79dfcec

Author: duduyi2013

fbgemm_gpu/src/ssd_split_embeddings_cache/ssd_split_table_batched_embeddings.cpp

@@ -478,7 +478,6 @@ void KVTensorWrapper::set_dram_db_wrapper(
 
 at::Tensor KVTensorWrapper::narrow(int64_t dim, int64_t start, int64_t length) {
   CHECK_EQ(dim, 0) << "Only narrow on dim 0 is supported";
-  CHECK_EQ(width_offset_, 0);
   if (db_) {
     CHECK_TRUE(db_ != nullptr);
     CHECK_GE(db_->get_max_D(), shape_[1]);