- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for getCcMajor (0.16 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_device_helper.cc
bool CanUseTensorCores(const RuntimeDevices &devices) { auto has_tensor_cores = [&](const DeviceNameUtils::ParsedName &device) { auto md = devices.GetGpuDeviceMetadata(device); return md ? md->getCcMajor() >= 7 : false; }; return llvm::all_of( llvm::make_filter_range(devices.device_names(), IsGpuDevice), has_tensor_cores); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 21 08:41:18 UTC 2022 - 2.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/device_util_test.cc
auto device_meta_1 = mlir::dyn_cast<mlir::TF::GpuDeviceMetadata>(devices_attr.get(gpu0)); ASSERT_NE(device_meta_1, nullptr); ASSERT_EQ(device_meta_1.getCcMajor(), 7); ASSERT_EQ(device_meta_1.getCcMinor(), 0); // If description is empty GPU devices added with an empty metadata. auto device_meta_2 = mlir::dyn_cast<mlir::UnitAttr>(devices_attr.get(gpu1));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.5K bytes - Viewed (0)