@@ -558,8 +558,8 @@ class DeviceCachingAllocator {
558
558
bool record_history = false ;
559
559
std::atomic<CreateContextFn> context_recorder_;
560
560
size_t alloc_trace_next = 0 ;
561
- bool alloc_trace_record_context = false ;
562
- size_t alloc_trace_max_entries = 1 ;
561
+ bool alloc_trace_record_context_ = false ;
562
+ size_t alloc_trace_max_entries_ = 1 ;
563
563
std::vector<TraceEntry>*
564
564
alloc_trace; // pointer because we need to intentionally leak this on
565
565
// deallocation it can hold references to Python state which
@@ -599,11 +599,10 @@ class DeviceCachingAllocator {
599
599
size_t alloc_trace_max_entries,
600
600
bool alloc_trace_record_context) {
601
601
std::unique_lock<std::recursive_mutex> lock (mutex);
602
- this ->record_history = enabled;
603
- this ->context_recorder_ .store (context_recorder);
604
- this ->alloc_trace_max_entries =
605
- std::max (size_t (1 ), alloc_trace_max_entries);
606
- this ->alloc_trace_record_context = alloc_trace_record_context;
602
+ record_history = enabled;
603
+ context_recorder_.store (context_recorder);
604
+ alloc_trace_max_entries_ = std::max (size_t (1 ), alloc_trace_max_entries);
605
+ alloc_trace_record_context_ = alloc_trace_record_context;
607
606
alloc_trace_next = 0 ;
608
607
alloc_trace->clear ();
609
608
}
@@ -1758,12 +1757,12 @@ class DeviceCachingAllocator {
1758
1757
addr,
1759
1758
size,
1760
1759
stream,
1761
- alloc_trace_record_context ? std::move (context) : nullptr );
1762
- if (alloc_trace->size () < alloc_trace_max_entries ) {
1760
+ alloc_trace_record_context_ ? std::move (context) : nullptr );
1761
+ if (alloc_trace->size () < alloc_trace_max_entries_ ) {
1763
1762
alloc_trace->emplace_back (te);
1764
1763
} else {
1765
1764
(*alloc_trace)[alloc_trace_next++] = te;
1766
- if (alloc_trace_next == alloc_trace_max_entries ) {
1765
+ if (alloc_trace_next == alloc_trace_max_entries_ ) {
1767
1766
alloc_trace_next = 0 ;
1768
1767
}
1769
1768
}
0 commit comments