Add testcases for cache c api

mindspore/ccsrc/minddata/dataset/api/datasets.cc

@@ -547,6 +547,7 @@ std::shared_ptr<ZipNode> Dataset::Zip(const std::vector<std::shared_ptr<Dataset>
 }
 Status Dataset::AddCacheOp(std::vector<std::shared_ptr<DatasetOp>> *node_ops) {
   if (cache_ != nullptr) {
+    RETURN_IF_NOT_OK(cache_->Build());
     std::shared_ptr<DatasetOp> cache_op;
     RETURN_IF_NOT_OK(cache_->CreateCacheOp(num_workers_, &cache_op));
     node_ops->push_back(cache_op);

mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/source/celeba_node.cc

@@ -31,7 +31,7 @@ namespace api {
 CelebANode::CelebANode(const std::string &dataset_dir, const std::string &usage,
                        const std::shared_ptr<SamplerObj> &sampler, const bool &decode,
                        const std::set<std::string> &extensions, const std::shared_ptr<DatasetCache> &cache)
-    : Dataset(cache),
+    : Dataset(std::move(cache)),
       dataset_dir_(dataset_dir),
       usage_(usage),
       sampler_(sampler),
@@ -60,6 +60,8 @@ std::vector<std::shared_ptr<DatasetOp>> CelebANode::Build() {
   RETURN_EMPTY_IF_ERROR(
     schema->AddColumn(ColDescriptor("attr", DataType(DataType::DE_UINT32), TensorImpl::kFlexible, 1)));
 
+  RETURN_EMPTY_IF_ERROR(AddCacheOp(&node_ops));
+
   node_ops.push_back(std::make_shared<CelebAOp>(num_workers_, rows_per_buffer_, dataset_dir_, connector_que_size_,
                                                 decode_, usage_, extensions_, std::move(schema),
                                                 std::move(sampler_->Build())));

mindspore/ccsrc/minddata/dataset/include/datasets.h

@@ -533,9 +533,10 @@ std::shared_ptr<VOCNode> VOC(const std::string &dataset_dir, const std::string &
 /// \param prefetch_sz optional prefetch size
 /// \return Shared pointer to DatasetCache. If error, nullptr is returned.
 std::shared_ptr<DatasetCache> CreateDatasetCache(session_id_type id, uint64_t mem_sz, bool spill,
-                                                 std::optional<std::string> hostname, std::optional<int32_t> port,
-                                                 std::optional<int32_t> num_connections,
-                                                 std::optional<int32_t> prefetch_sz);
+                                                 std::optional<std::string> hostname = std::nullopt,
+                                                 std::optional<int32_t> port = std::nullopt,
+                                                 std::optional<int32_t> num_connections = std::nullopt,
+                                                 std::optional<int32_t> prefetch_sz = std::nullopt);
 #endif
 
 /// \brief Function to create a ZipNode

tests/ut/cpp/dataset/CMakeLists.txt

@@ -119,6 +119,7 @@ SET(DE_UT_SRCS
         c_api_datasets_test.cc
         c_api_dataset_iterator_test.cc
         c_api_text_vocab_test.cc
+        c_api_cache_test.cc
         tensor_op_fusion_pass_test.cc
         sliding_window_op_test.cc
         epoch_ctrl_op_test.cc