Skip to content

Commit 7df9705

Browse files
yaxanYazan Masoud
andauthored
Fix centml.compiler.prediction ModuleNotFoundError (#75)
* Adds missing __init__.py for prediction folder * Remove unused variable --------- Co-authored-by: Yazan Masoud <yazanmasoud@Yazans-MacBook-Pro.local>
1 parent 0054da5 commit 7df9705

File tree

2 files changed

+1
-1
lines changed

2 files changed

+1
-1
lines changed

centml/compiler/prediction/__init__.py

Whitespace-only changes.

centml/compiler/prediction/profiler.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,7 @@ def get_time_or_profile(key, inp_shapes, operation, *args, **kwargs):
8383

8484
if self.data_collection_mode and t is None:
8585
with torch.profiler.profile(activities=[torch.profiler.ProfilerActivity.CUDA]) as prof:
86-
result = operation(*args, **kwargs)
86+
operation(*args, **kwargs)
8787
event_time_total = 0
8888
for event in prof.key_averages():
8989
event_time_total += event.cuda_time_total

0 commit comments

Comments
 (0)