From 0edaaa0c54ef8a1e8cd9a6e912ac0732cae35cf6 Mon Sep 17 00:00:00 2001
From: Roger Waleffe <rogerwaleffe@Rogers-MBP.attlocal.net>
Date: Tue, 21 Nov 2023 13:43:22 -0600
Subject: [PATCH] unique feature timing

---
 src/cpp/src/data/dataloader.cpp | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/cpp/src/data/dataloader.cpp b/src/cpp/src/data/dataloader.cpp
index f52f1ee5..5972b837 100644
--- a/src/cpp/src/data/dataloader.cpp
+++ b/src/cpp/src/data/dataloader.cpp
@@ -665,6 +665,8 @@ void DataLoader::loadCPUParameters(shared_ptr<Batch> batch) {
 //                                  << batch->sub_batches_[i]->unique_node_indices_.device() << "\n";
                     }
 
+                    Timer t = new Timer(false);
+                    t.start();
 //                    std::cout << "cat\n";
                     torch::Tensor all_unique_nodes = torch::cat({all_unique_nodes_vec}, 0);
 //                    std::cout << all_unique_nodes.sizes() << "\n";
@@ -676,6 +678,8 @@ void DataLoader::loadCPUParameters(shared_ptr<Batch> batch) {
 //                    std::cout << inverse.sizes() << " " << inverse.device() << "\n";
 //                    std::cout << unique_features.sizes() << " " << unique_features.device() << "\n";
                     std::cout<<unique_indices.size(0)<<" vs " <<all_unique_nodes.size(0)<<"\n";
+                    t.stop();
+                    std::cout<<"uniques: "<<t.getDuration()<<"\n";
 
 //                    std::cout << "end cat\n";
                     int count = 0;