projects
/
profiler-torch.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Decorate as high as possible in the class hierarchy.
[profiler-torch.git]
/
profiler.lua
diff --git
a/profiler.lua
b/profiler.lua
index
f57c9d8
..
77461b3
100644
(file)
--- a/
profiler.lua
+++ b/
profiler.lua
@@
-64,15
+64,20
@@
function profiler.decorate(model, functionsToDecorate)
-- save models anymore.
if rawget(model, name) then
-- save models anymore.
if rawget(model, name) then
- error('We decorate the class, not the objects, and there is a `'
+ error('We decorate the class
es
, not the objects, and there is a `'
.. name
.. '\' function in '
.. tostring(model))
end
.. name
.. '\' function in '
.. tostring(model))
end
- local toDecorate =
getmetatable(model)
+ local toDecorate =
model
- if toDecorate[name] and not toDecorate[nameOrig] then
+ while not rawget(toDecorate, name) do
+ toDecorate = getmetatable(toDecorate)
+ end
+
+ if not toDecorate[nameOrig] then
+ print('Decorating ' .. toDecorate.__typename .. '.' .. name)
toDecorate[nameOrig] = toDecorate[name]
toDecorate[name] = function(self, ...)
local startTime = sys.clock()
toDecorate[nameOrig] = toDecorate[name]
toDecorate[name] = function(self, ...)
local startTime = sys.clock()