Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for SynchronizeAllActivity (0.26 sec)

  1. tensorflow/c/experimental/stream_executor/stream_executor_test_util.cc

                        uint64_t size, TF_Status* const status) {}
    void BlockHostForEvent(const SP_Device* const device, SP_Event event,
                           TF_Status* const status) {}
    void SynchronizeAllActivity(const SP_Device* const device,
                                TF_Status* const status) {}
    TF_Bool HostCallback(const SP_Device* const device, SP_Stream stream,
                         SE_StatusCallbackFn const callback_fn,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 21 20:09:00 UTC 2023
    - 8.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/xla_device_compiler_client.cc

    }
    
    void XlaDeviceCompilerClient::WaitForProgramsToFinish() {
      if (client_ == nullptr) return;
    
      for (auto* executor : client_->backend().stream_executors()) {
        bool ok = executor->SynchronizeAllActivity();
        if (!ok) {
          LOG(ERROR) << "Error synchronizing activity while waiting for all "
                        "programs to complete";
        }
      }
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 09:53:30 UTC 2024
    - 4.1K bytes
    - Viewed (0)
  3. tensorflow/c/experimental/stream_executor/stream_executor_test.cc

      ASSERT_FALSE(block_host_until_done_called);
      TF_ASSERT_OK(stream->BlockHostUntilDone());
      ASSERT_TRUE(block_host_until_done_called);
    }
    
    TEST_F(StreamExecutorTest, SynchronizeAllActivity) {
      static bool synchronize_all_called = false;
      se_.synchronize_all_activity = [](const SP_Device* const device,
                                        TF_Status* const status) {
        TF_SetStatus(status, TF_OK, "");
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 20 19:54:04 UTC 2024
    - 26.5K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/xla_device_context.cc

      tf_stats.largest_free_block_bytes = se_stats->largest_free_block_bytes;
      return tf_stats;
    }
    
    bool XlaDeviceAllocator::ClearStats() {
      if (!stream_executor_->SynchronizeAllActivity()) {
        return false;
      }
      return stream_executor_->ClearAllocatorStats();
    }
    
    XlaDeviceContext::XlaDeviceContext(
        std::shared_ptr<se::Stream> compute_stream,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 16 00:36:08 UTC 2024
    - 12.7K bytes
    - Viewed (0)
  5. tensorflow/c/experimental/stream_executor/stream_executor.cc

          stats.bytes_reservable_limit = c_stats.bytes_reservable_limit;
        }
        stats.largest_free_block_bytes = c_stats.largest_free_block_bytes;
        return stats;
      }
      bool SynchronizeAllActivity() override {
        OwnedTFStatus c_status(TF_NewStatus());
        stream_executor_->synchronize_all_activity(&device_, c_status.get());
        if (TF_GetCode(c_status.get()) != TF_OK) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Jun 14 07:39:19 UTC 2024
    - 27.1K bytes
    - Viewed (0)
Back to top