mirror of
https://github.com/leejet/stable-diffusion.cpp.git
synced 2026-07-28 13:50:53 -05:00
25 lines
714 B
C++
25 lines
714 B
C++
#ifndef __SD_CORE_LAYER_SPLIT_PARTITION_H__
|
|
#define __SD_CORE_LAYER_SPLIT_PARTITION_H__
|
|
|
|
#include <map>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include "ggml-backend.h"
|
|
#include "ggml.h"
|
|
|
|
namespace sd {
|
|
|
|
std::string layer_split_backend_device_display_name(ggml_backend_t backend);
|
|
int layer_split_tensor_block_index(const std::string& name);
|
|
|
|
std::vector<std::map<std::string, ggml_tensor*>> partition_layer_split_tensors(
|
|
const std::string& desc,
|
|
const std::map<std::string, ggml_tensor*>& tensors,
|
|
const std::map<std::string, ggml_tensor*>& split_tensors,
|
|
const std::vector<ggml_backend_t>& backends);
|
|
|
|
} // namespace sd
|
|
|
|
#endif // __SD_CORE_LAYER_SPLIT_PARTITION_H__
|