1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41
   42
   43
   44
   45
   46
   47
   48
   49
   50
   51
   52
   53
   54
   55
   56
   57
   58
   59
   60
   61
   62
   63
   64
   65
   66
   67
   68
   69


gpu / command_buffer / client / dawn_client_memory_transfer_service.h [blame]

// Copyright 2019 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef GPU_COMMAND_BUFFER_CLIENT_DAWN_CLIENT_MEMORY_TRANSFER_SERVICE_H_
#define GPU_COMMAND_BUFFER_CLIENT_DAWN_CLIENT_MEMORY_TRANSFER_SERVICE_H_

#include <dawn/wire/WireClient.h>
#include <vector>

#include "base/memory/raw_ptr.h"

namespace gpu {

class CommandBufferHelper;
class MappedMemoryManager;

namespace webgpu {

struct MemoryTransferHandle;

class DawnClientMemoryTransferService
    : public dawn::wire::client::MemoryTransferService {
 public:
  DawnClientMemoryTransferService(MappedMemoryManager* mapped_memory);
  ~DawnClientMemoryTransferService() override;

  // Create a handle for reading shared memory data.
  // This may fail and return nullptr.
  ReadHandle* CreateReadHandle(size_t size) override;

  // Create a handle for writing shared memory data.
  // This may fail and return nullptr.
  WriteHandle* CreateWriteHandle(size_t size) override;

  // Free shared memory allocations after the next token passes on the GPU
  // process.
  void FreeHandles(CommandBufferHelper* helper);

  void Disconnect();

 private:
  class ReadHandleImpl;
  class WriteHandleImpl;

  // Allocate a shared memory handle for the memory transfer.
  void* AllocateHandle(size_t size, MemoryTransferHandle* handle);

  // Mark a shared memory allocation as free. This should not be called more
  // than once per block.
  void MarkHandleFree(void* ptr);

  // Found dangling on `linux-rel` in
  // `gpu_tests.context_lost_integration_test.ContextLostIntegrationTest.
  // ContextLost_WebGPUStressRequestDeviceAndRemoveLoop`
  raw_ptr<MappedMemoryManager, DanglingUntriaged> mapped_memory_;

  // Pointers to memory allocated by the MappedMemoryManager to free after
  // the next Flush.
  std::vector<raw_ptr<void, VectorExperimental>> free_blocks_;

  // If disconnected, new handle creation always returns null.
  bool disconnected_ = false;
};

}  // namespace webgpu
}  // namespace gpu

#endif  // GPU_COMMAND_BUFFER_CLIENT_DAWN_CLIENT_MEMORY_TRANSFER_SERVICE_H_