NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-04-20 06:48:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fail gracefully when PTX is not found for a device.

Browse Source
This commit is contained in:
Allison Vacanti
2021-03-05 10:50:57 -05:00
parent b1269b2f7f
commit 27eb2b81ff
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
nvbench/device_info.cuh
View File

@@ -210,6 +210,7 @@ __global__ void noop_kernel()
inline const auto noop_kernel_ptr = &noop_kernel<void>;
[[nodiscard]] inline int get_ptx_version(int dev_id)
try
{
nvbench::detail::device_scope _{dev_id};
cudaFuncAttributes attr{};
@@ -217,6 +218,11 @@ inline const auto noop_kernel_ptr = &noop_kernel<void>;
cudaFuncGetAttributes(&attr, nvbench::detail::noop_kernel_ptr));
return attr.ptxVersion * 10;
}
catch(...)
{ // Fail gracefully when no appropriate PTX is found for this device.
return -1;
}
} // namespace detail
} // namespace nvbench