This repository was archived by the owner on Apr 28, 2023. It is now read-only.
File tree Expand file tree Collapse file tree 3 files changed +7
-5
lines changed Expand file tree Collapse file tree 3 files changed +7
-5
lines changed Original file line number Diff line number Diff line change @@ -86,7 +86,7 @@ class CpuTcExecutor : public ::tc::TcExecutor {
86
86
const std::vector<void *>& outputs) const ;
87
87
88
88
bool hasRuntimeCompiledFunction () override {
89
- return rtcFun .get () != nullptr ;
89
+ return rtcFunction .get () != nullptr ;
90
90
}
91
91
92
92
// It is necessary to clear the RTC manually because it can throw and we
@@ -95,7 +95,7 @@ class CpuTcExecutor : public ::tc::TcExecutor {
95
95
if (!hasRuntimeCompiledFunction ()) {
96
96
return ;
97
97
}
98
- rtcFun ->clear ();
98
+ rtcFunction ->clear ();
99
99
}
100
100
101
101
std::string kernelName () const {
@@ -110,7 +110,7 @@ class CpuTcExecutor : public ::tc::TcExecutor {
110
110
std::string cpuSource;
111
111
112
112
protected:
113
- std::shared_ptr<CpuRTCFunction> rtcFun ;
113
+ std::shared_ptr<CpuRTCFunction> rtcFunction ;
114
114
};
115
115
116
116
} // namespace tc
Original file line number Diff line number Diff line change @@ -114,7 +114,7 @@ size_t ExecutionEngine<ExecutorType>::compile(
114
114
return handle;
115
115
}
116
116
117
- // Steal ExecutorType and give it back under lock
117
+ // Steal the executor instance and give it back under lock.
118
118
// Run outside of lock on owning ExecutorType.
119
119
template <typename ExecutorType>
120
120
Duration ExecutionEngine<ExecutorType>::run(
@@ -188,6 +188,7 @@ void ExecutionEngine<ExecutorType>::uncheckedRun(
188
188
}
189
189
}
190
190
}
191
+
191
192
// Clear the underlying RTC object and executor under lock.
192
193
template <typename ExecutorType>
193
194
void ExecutionEngine<ExecutorType>::clear(size_t handle) {
Original file line number Diff line number Diff line change @@ -39,7 +39,8 @@ Duration CpuTcExecutor::run(
39
39
const std::vector<const DLTensor*>& inputs,
40
40
const std::vector<DLTensor*>& outputs,
41
41
bool profile) const {
42
- CHECK (rtcFun) << " Can't launch uncompiled: " << executionInfo_.kernelName ;
42
+ CHECK (rtcFunction) << " Can't launch uncompiled: "
43
+ << executionInfo_.kernelName ;
43
44
CHECK_NE (executionInfo_.options , " " );
44
45
checkSizesAndStridesAreCompliant (
45
46
inputs, executionInfo_.inputsInfo , halideComponents_.getDef ().params ());
You can’t perform that action at this time.
0 commit comments