blender/intern/cycles/util/util_stats.h
Sergey Sharybin fbed2047c8 Fix wrong track of the memory when doing device vector resize before freeing it
This is rather legit case which happens i.e. when having persistent images enabled
and session is updating the lookup tables.

Now device_memory keeps track of amount of memory being allocated on the device,
which makes freeing using the proper allocated size, not the CPU side buffer
size.
2014-09-04 17:25:12 +06:00

44 lines
1013 B
C++

/*
* Copyright 2011-2013 Blender Foundation
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License
*/
#ifndef __UTIL_STATS_H__
#define __UTIL_STATS_H__
CCL_NAMESPACE_BEGIN
class Stats {
public:
Stats() : mem_used(0), mem_peak(0) {}
void mem_alloc(size_t size) {
mem_used += size;
if(mem_used > mem_peak)
mem_peak = mem_used;
}
void mem_free(size_t size) {
assert(mem_used >= size);
mem_used -= size;
}
size_t mem_used;
size_t mem_peak;
};
CCL_NAMESPACE_END
#endif /* __UTIL_STATS_H__ */