Searched refs:device_cap (Results 1 – 3 of 3) sorted by relevance
30 uint32 GetFmaMaxThroughputPerSMPerCycle(const DeviceCapabilities& device_cap) { in GetFmaMaxThroughputPerSMPerCycle() argument33 switch (device_cap.compute_capability().major()) { in GetFmaMaxThroughputPerSMPerCycle()48 if (device_cap.compute_capability().minor() > 0) { in GetFmaMaxThroughputPerSMPerCycle()63 if (device_cap.compute_capability().minor() >= 6) { in GetFmaMaxThroughputPerSMPerCycle()83 double GetFlopMaxThroughputPerSM(const DeviceCapabilities& device_cap) { in GetFlopMaxThroughputPerSM() argument85 return GetFmaMaxThroughputPerSMPerCycle(device_cap) * 2 * in GetFlopMaxThroughputPerSM()86 device_cap.clock_rate_in_ghz(); in GetFlopMaxThroughputPerSM()89 absl::string_view GpuModelName(const DeviceCapabilities& device_cap) { in GpuModelName() argument90 switch (device_cap.compute_capability().major()) { in GpuModelName()100 if (device_cap.compute_capability().minor() < 5) { in GpuModelName()
27 double GetFlopMaxThroughputPerSM(const DeviceCapabilities& device_cap);30 absl::string_view GpuModelName(const DeviceCapabilities& device_cap);
342 let device_cap = VirtioPciCap::new( in add_settings_pci_capabilities() localVariable349 .add_capability(&device_cap) in add_settings_pci_capabilities()