Ejemplo n.º 1
0
    def test_memory_manager_backend(self):
        # Set the global GPU memory manager backend. By default PYTORCH.
        ME.set_memory_manager_backend(MemoryManagerBackend.PYTORCH)
        ME.set_memory_manager_backend(MemoryManagerBackend.CUDA)

        # Create a coords man with the specified GPU memory manager backend.
        # No effect with CPU_ONLY build
        cm = CoordsManager(memory_manager_backend=MemoryManagerBackend.CUDA, D=2)
        cm = CoordsManager(memory_manager_backend=MemoryManagerBackend.PYTORCH, D=2)