pub_soft/dgl
3
0
Fork 0
mirror of https://github.com/dmlc/dgl.git synced 2026-06-04 19:44:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

[GraphBolt] Remove assertions from CachePolicy. (#7653)

Browse Source
This commit is contained in:
Muhammed Fatih BALIN
2024-08-04 16:25:16 -04:00
committed by GitHub
parent d87b758216
commit 2babaf9d66
1 changed files with 4 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
graphbolt/src/cache_policy.h
View File

@@ -79,18 +79,15 @@ struct CacheKey {
}
CacheKey& StartRead() {
TORCH_CHECK(reference_count_ >= 0);
TORCH_CHECK(reference_count_++ < std::numeric_limits<int16_t>::max());
++reference_count_;
return *this;
}
template <bool write>
CacheKey& EndUse() {
if constexpr (write) {
TORCH_CHECK(reference_count_ == -1);
++reference_count_;
} else {
TORCH_CHECK(reference_count_ > 0);
--reference_count_;
}
return *this;
@@ -435,7 +432,7 @@ class SieveCachePolicy : public BaseCachePolicy {
if (hand_ == queue_.begin()) hand_ = queue_.end();
--hand_;
}
TORCH_CHECK(key_to_cache_key_.erase(hand_->getKey()));
key_to_cache_key_.erase(hand_->getKey());
const auto pos = hand_->getPos();
const auto temp = hand_;
if (hand_ == queue_.begin()) {
@@ -535,7 +532,7 @@ class LruCachePolicy : public BaseCachePolicy {
MoveToFront(queue_, queue_, it);
}
const auto& cache_key = queue_.back();
TORCH_CHECK(key_to_cache_key_.erase(cache_key.getKey()));
key_to_cache_key_.erase(cache_key.getKey());
const auto pos = cache_key.getPos();
queue_.pop_back();
return pos;
@@ -629,7 +626,7 @@ class ClockCachePolicy : public BaseCachePolicy {
std::advance(it, -1);
MoveToFront(queue_, queue_, it);
} else {
TORCH_CHECK(key_to_cache_key_.erase(cache_key.getKey()));
key_to_cache_key_.erase(cache_key.getKey());
const auto evicted_pos = cache_key.getPos();
queue_.pop_back();
return evicted_pos;