resources.hpp

Go to the documentation of this file.

// Copyright 2025 Eric Hermosis
//
// This file is part of the Tannic Tensor Library.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
// 
 
#ifndef RESOURCES_HPP
#define RESOURCES_HPP 
 
#include <cstdint>
#include <cstddef>
#include <span>
#include <vector>
#include <variant>   
 
namespace tannic {
 
class Host {
public: 
 
    Host() = default;
 
    void* allocate(std::size_t nbytes) const;
 
    void deallocate(void* address, std::size_t nbytes) const; 
 
    int id() const { return -1; } 
 
    bool pageable() const {
        return pageable_;
    }
 
    bool pinned() const {
        return pinned_;
    }
 
private:
    bool pageable_ = true; 
    bool pinned_ = false;
};  
 
 
class Devices {
private:
    int count_ = 0; 
    Devices();
 
public: 
    Devices(const Devices&) = delete;
    Devices& operator=(const Devices&) = delete;
    Devices(Devices&&) = delete;
    Devices& operator=(Devices&&) = delete; 
    ~Devices() = default;
 
    static Devices& instance() {
        static Devices instance;
        return instance;
    }
 
    static int count() {
        Devices& devices = instance();
        return devices.count_;
    } 
}; 
 
 
class Device {
public: 
 
    Device(int id = 0, bool blocking = false); 
 
    void* allocate(std::size_t nbytes) const;
 
 
    void deallocate(void* address, std::size_t nbytes) const; 
 
    int id() const noexcept { 
        return id_; 
    }
    
    bool blocking() const {
        return blocking_;
    }
 
private:
    int id_ = 0;
    bool blocking_ = false;
};
 
 
using Environment = std::variant<Host, Device>;
 
} // namespace tannic
 
#endif // RESOURCES_HPP