Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for getCcMajor (0.16 sec)

  1. 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)
  2. 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)
Back to top