Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 68 for r2devices (0.24 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
  7. 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)
  8. 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)
  9. 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)
  10. 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)
Back to top