Home / Class/ ReplayBufferCleaner Class — pytorch Architecture

ReplayBufferCleaner Class — pytorch Architecture

Architecture documentation for the ReplayBufferCleaner class in mps_test_allocator.cpp from the pytorch codebase.

Class cpp

Entity Profile

Source Code

aten/src/ATen/test/mps_test_allocator.cpp lines 8–15

class ReplayBufferCleaner : virtual public at::mps::IMpsAllocatorCallback {
    public:
    void executeMPSAllocatorCallback(void* ptr, EventType event) override {
     if (event == EventType::ALLOCATION_FAILED) {
        callback_action();
     }
    }
};

Source

Analyze Your Own Codebase

Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.

Try Supermodel Free