- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 68 for r2devices (0.24 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util.cc
absl::StrCat("found ", kDeviceTPUSystem, " devices with conflicting ", attribute, "s '", a, "' and '", b, "'")); } // Find TPU_SYSTEM:0 devices in `devices`. If multiple TPU_SYSTEM devices are // found, the first one lexicographically is returned. If no TPU_SYSTEM device // is found or if there are multiple TPU_SYSTEM devices with different jobs or // replicas, a failure will be returned.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jun 10 20:10:40 UTC 2024 - 32.8K bytes - Viewed (0) -
pkg/kubelet/cm/devicemanager/manager.go
// Check if registered resource has healthy devices if healthyDevices.Len() == 0 { return nil, fmt.Errorf("no healthy devices present; cannot allocate unhealthy devices %s", resource) } // Check if all the previously allocated devices are healthy if !healthyDevices.IsSuperset(devices) { return nil, fmt.Errorf("previously allocated devices are no longer healthy; cannot allocate unhealthy devices %s", resource) }
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Mon Apr 15 12:01:56 UTC 2024 - 43K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util.h
// Finds the TPU compilation device and execution devices from `devices` for a // TPU computation subgraph. Compilation device is determined from looking up // all TPU_SYSTEM:0 devices and choosing the CPU device associated to the first // TPU_SYSTEM device sorted lexicographically by replica and task. Execution // devices are determined by looking up all TPU devices associated with each
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 11.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util_test.cc
std::vector<std::string> devices{ "/job:localhost/replica:0/task:0/device:CPU:0"}; devices.reserve(num_tasks * num_devices_per_task + num_tasks + 1); for (int task = 0; task < num_tasks; ++task) { devices.push_back( llvm::formatv("/job:worker/replica:0/task:{0}/device:CPU:0", task) .str()); devices.push_back(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 46.8K bytes - Viewed (0) -
pkg/kubelet/apis/podresources/server_v1_test.go
{ Name: containerName, Devices: []*podresourcesapi.ContainerDevices{}, DynamicResources: []*podresourcesapi.DynamicResource{}, }, }, }, }, }, }, { desc: "pod with devices", pods: pods, devices: devs, cpus: cpus, memory: memory,
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Wed Apr 24 18:25:29 UTC 2024 - 45.9K bytes - Viewed (0) -
pkg/kubelet/cm/devicemanager/pod_devices_test.go
for devID, plugInfo := range contDevices { nodes := plugInfo.GetTopology().GetNodes() require.Equal(t, len(nodes), len(devices), "Incorrect container devices: %v - %v (nodes %v)", devices, contDevices, nodes) for _, node := range plugInfo.GetTopology().GetNodes() { dev, ok := devices[node.ID] require.True(t, ok, "NUMA id %v doesn't exist in result", node.ID)
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Wed Apr 24 18:25:29 UTC 2024 - 10.3K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_tpu_device.cc
public: Status ListPhysicalDevices(std::vector<string>* devices) override; Status CreateDevices(const SessionOptions& options, const string& name_prefix, std::vector<std::unique_ptr<Device>>* devices) override; }; Status TpuNodeDeviceFactory::ListPhysicalDevices(std::vector<string>* devices) { tpu::TpuPlatformInterface* platform =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 20.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_device_propagation.cc
lhs_device_attr.getValue() == rhs_device_attr.getValue()); }; // Check if tf_executor.NextIteration.Source/tf_executor.NextIteration.Sink // pair has matching devices or no devices. if (auto source = llvm::dyn_cast<tf_executor::NextIterationSourceOp>(op)) { return ops_have_same_device(source, source.GetSink()); } else if (auto sink = llvm::dyn_cast<tf_executor::NextIterationSinkOp>(op)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc
if (execute_arg_to_outer_args.empty()) return false; // Extract the replicated devices. auto devices_attr = replicate.getDevices(); if (!devices_attr) return false; auto device_map = devices_attr.value(); llvm::SmallDenseMap<llvm::StringRef, llvm::SmallVector<StringRef, 4>> devices; devices.reserve(device_map.size()); for (auto it : device_map) { auto device_alias = it.getName().strref();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_rewrite_pass.cc
return tensorflow::WrapOpInLaunch(builder, compile_op.getLoc(), compile_op, compilation_device); } // Assigns explicit devices to replicate op. An aliased device is created per // core, and all replica devices per core are grouped together. void AssignDevicesToReplicate( tf_device::ReplicateOp replicate, llvm::ArrayRef<llvm::SmallVector<tensorflow::TPUDeviceAndHost, 8>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 29.7K bytes - Viewed (0)