-
Notifications
You must be signed in to change notification settings - Fork 5
Open
Labels
Description
Currently we query the device/compute capability to figure out
- which data types are supported by tensor cores ()
GPUInspector.jl/src/peakflops_gpu.jl
Line 61 in f5e17e5
if cap == v"8.0.0" - the number of cuda cores ()
GPUInspector.jl/src/gpuinfo.jl
Line 248 in f5e17e5
function ncudacores(major, minor, mp) - etc.?
Can we be more general with this?