Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 30 of 54 for set_device (0.15 sec)

  1. pkg/volume/csi/csi_block_test.go

    	if err != nil {
    		t.Fatalf("failed to setup VolumeAttachment: %v", err)
    	}
    	t.Log("created attachment ", attachID)
    
    	stagingPath, err := csiMapper.SetUpDevice()
    	if err != nil {
    		t.Fatalf("mapper failed to SetupDevice: %v", err)
    	}
    
    	// Check if NodeStageVolume staged to the right path
    	svols := csiMapper.csiClient.(*fakeCsiDriverClient).nodeClient.GetNodeStagedVolumes()
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Mon Sep 11 06:07:40 UTC 2023
    - 26.5K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_test_util.h

    // `send_device`, `recv_device`, and `send_device_incarnation` set.
    TFE_Op* SendOp(TFE_Context* ctx, TFE_TensorHandle* in,
                   const std::string& op_name, const std::string& send_device,
                   const std::string& recv_device,
                   tensorflow::uint64 send_device_incarnation);
    
    // Return a RecvOp op `op_name` with the attributes `send_device`,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Jul 17 23:43:59 UTC 2023
    - 7.7K bytes
    - Viewed (0)
  3. pkg/volume/csi/csi_block.go

    		return "", err
    	}
    
    	return publishPath, nil
    }
    
    // SetUpDevice ensures the device is attached returns path where the device is located.
    func (m *csiBlockMapper) SetUpDevice() (string, error) {
    	klog.V(4).Infof(log("blockMapper.SetUpDevice called"))
    
    	// Get csiSource from spec
    	if m.spec == nil {
    		return "", errors.New(log("blockMapper.SetUpDevice spec is nil"))
    	}
    
    	csiSource, err := getCSISourceFromSpec(m.spec)
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Mon Sep 11 06:07:40 UTC 2023
    - 20.1K bytes
    - Viewed (0)
  4. tensorflow/c/experimental/stream_executor/stream_executor_internal.h

          event_handle_ = nullptr;
        }
      }
    
      SP_Event Handle() { return event_handle_; }
    
     private:
      SP_Device* device_;
      SP_StreamExecutor* stream_executor_;
      SP_Event event_handle_;
    };
    
    class CStream : public StreamCommon {
     public:
      CStream(SP_Device* device, SP_StreamExecutor* stream_executor,
              StreamExecutor* executor)
          : StreamCommon(executor),
            device_(device),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Jun 14 07:39:19 UTC 2024
    - 8K bytes
    - Viewed (0)
  5. tensorflow/compiler/jit/xla_platform_info_test.cc

    };
    
    #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
    TEST_F(XlaPlatformInfoTest, BuildXlaDeviceCompilerXlaDeviceMetadata) {
      device_setup_.AddDevicesAndSetUp({DEVICE_XLA_GPU});
    
      Device* device = device_setup_.GetDevice(DEVICE_XLA_GPU);
      const XlaDevice::Metadata* metadata = nullptr;
      TF_CHECK_OK(XlaDevice::GetMetadataFromDevice(device, &metadata));
      XlaPlatformInfo platform_info = XlaPlatformInfoFromDevice(device);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sun Jan 14 15:17:12 UTC 2024
    - 13.6K bytes
    - Viewed (0)
  6. tensorflow/c/eager/c_api_test_util.cc

      TF_DeleteStatus(status);
    
      TFE_OpSetAttrType(op, "T", TFE_TensorHandleDataType(in));
      TFE_OpSetAttrString(op, "tensor_name", "dummy", 5);
      TFE_OpSetAttrString(op, "send_device", send_device.c_str(),
                          send_device.size());
      TFE_OpSetAttrString(op, "recv_device", recv_device.c_str(),
                          recv_device.size());
      TFE_OpSetAttrInt(op, "send_device_incarnation", send_device_incarnation);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 22:37:46 UTC 2024
    - 23.5K bytes
    - Viewed (0)
  7. tensorflow/compiler/jit/xla_compiler_options_util_test.cc

      }
    
      DeviceSetup device_setup_;
    };
    
    TEST_F(XlaCompilerOptionsTest, PjRtOptionsXlaDevice) {
      device_setup_.AddDevicesAndSetUp({DEVICE_XLA_GPU});
      Device* device = device_setup_.GetDevice(DEVICE_XLA_GPU);
      DeviceType compilation_device_type = DeviceType(DEVICE_GPU_XLA_JIT);
    
      se::Platform::Id platform_id = nullptr;
      auto xla_device_metadata = CreateXlaDeviceMetadata(compilation_device_type);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Dec 29 01:41:20 UTC 2023
    - 14.8K bytes
    - Viewed (0)
  8. tensorflow/compiler/jit/pjrt_compile_util_test.cc

    TEST(PjrtCompileUtilTest, CompileToPjRtLoadedExecutable) {
      DeviceSetup device_setup;
      TF_ASSERT_OK_AND_ASSIGN(auto fdef, SampleFuntionAddXY("foo"));
      device_setup.AddDevicesAndSetUp({DEVICE_GPU}, fdef);
    
      Device* device = device_setup.GetDevice(DEVICE_GPU);
      const XlaPlatformInfo platform_info = XlaPlatformInfoFromDevice(device);
    
      NameAttrList function;
      function.set_name("foo");
    
      ResourceMgr resource_mgr("");
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Aug 21 23:21:57 UTC 2023
    - 4.7K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/transforms/launch_to_device_attribute.cc

            return WalkResult::advance();
          } else if (device_str_attr.getValue() != launch.getDevice()) {
            return launch.emitOpError()
                   << "inner op has conflicting 'device' attribute, "
                      "got '"
                   << device_str_attr.getValue() << "' but expected '"
                   << launch.getDevice() << "'";
          }
        } else {
          return launch.emitOpError()
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.5K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/ir/tf_remaining_ops.cc

      return getKey().str();
    }
    
    namespace {
    std::string GetRendezvousKey(const std::string& send_device,
                                 const uint64_t send_device_incarnation,
                                 const std::string& recv_device,
                                 const std::string& tensor_name) {
      return absl::StrCat(send_device, ";", send_device_incarnation, ";",
                          recv_device, ";", tensor_name);
    }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 14 20:05:58 UTC 2023
    - 7.6K bytes
    - Viewed (0)
Back to top