@@ -329,7 +329,7 @@ class Context {
|
||||
const profile_vector_t profile_vector = GetProfiles(group_index);
|
||||
for (auto& tuple : profile_vector) {
|
||||
// Wait for stop packet to complete
|
||||
const uint64_t timeout = UINT64_MAX;
|
||||
const uint64_t timeout = timeout_;
|
||||
bool complete = false;
|
||||
while (!complete) {
|
||||
const hsa_signal_value_t signal_value = hsa_signal_wait_scacquire(tuple.completion_signal, HSA_SIGNAL_CONDITION_LT, 1, timeout,
|
||||
@@ -372,6 +372,10 @@ class Context {
|
||||
}
|
||||
}
|
||||
|
||||
static void SetTimeout(uint64_t timeout) {
|
||||
timeout_ = timeout;
|
||||
}
|
||||
|
||||
private:
|
||||
// Getting profling packets
|
||||
profile_vector_t GetProfiles(const uint32_t& index) {
|
||||
@@ -469,6 +473,9 @@ class Context {
|
||||
return info;
|
||||
}
|
||||
|
||||
// Profiling data waiting timeout
|
||||
static uint64_t timeout_;
|
||||
|
||||
// GPU handel
|
||||
const hsa_agent_t agent_;
|
||||
const util::AgentInfo* agent_info_;
|
||||
|
||||
Odkázat v novém úkolu
Zablokovat Uživatele