mirror of
https://github.com/GraphiteEditor/Graphite.git
synced 2026-09-17 15:28:04 +08:00
f2b911b358
* Implement skeleton for graphene-cli * Configure gpu surface on non wasm32 targets * Create window with full hd size * Create window using the graphen-cli * Use window size for surface creation * Reuse surface configuration * Reduce window size for native applications to 800x600 * Add compute pipeline test * Poll wgpu execution externally * Remove cache node after texture upload * Add profiling instructions * Add more debug markers * Evaluate extract node before flattening the network * Reenable hue saturation node for compilation * Make hue saturation node work on the gpu + make f32 default for user inputs * Add version of test files without caching * Only dispatch each workgroup not pixel * ICE * Add quantization to gpu code * Fix quantization * Load images at graph runtime * Fix quantization calculation * Feature gate quantization * Use git version of autoquant * Add license to `graphene-cli` * Fix graphene-cli test case * Ignore tests on non unix platforms * Fix flattening test
50 lines
1.6 KiB
Rust
50 lines
1.6 KiB
Rust
use std::{collections::HashMap, sync::Arc, sync::RwLock};
|
|
|
|
use gpu_compiler_bin_wrapper::CompileRequest;
|
|
use tower_http::cors::CorsLayer;
|
|
|
|
use axum::{
|
|
extract::{Json, State},
|
|
http::StatusCode,
|
|
routing::{get, post},
|
|
Router,
|
|
};
|
|
|
|
struct AppState {
|
|
compile_dir: tempfile::TempDir,
|
|
cache: RwLock<HashMap<CompileRequest, Result<Vec<u8>, StatusCode>>>,
|
|
}
|
|
|
|
#[tokio::main]
|
|
async fn main() {
|
|
let shared_state = Arc::new(AppState {
|
|
compile_dir: tempfile::tempdir().expect("failed to create tempdir"),
|
|
cache: Default::default(),
|
|
});
|
|
|
|
// build our application with a single route
|
|
let app = Router::new()
|
|
.route("/", get(|| async { "Hello from compilation server!" }))
|
|
.route("/compile", get(|| async { "Supported targets: spirv" }))
|
|
.route("/compile/spirv", post(post_compile_spirv))
|
|
.layer(CorsLayer::permissive())
|
|
.with_state(shared_state);
|
|
|
|
// run it with hyper on localhost:3000
|
|
axum::Server::bind(&"0.0.0.0:3000".parse().unwrap()).serve(app.into_make_service()).await.unwrap();
|
|
}
|
|
|
|
async fn post_compile_spirv(State(state): State<Arc<AppState>>, Json(compile_request): Json<CompileRequest>) -> Result<Vec<u8>, StatusCode> {
|
|
if let Some(result) = state.cache.read().unwrap().get(&compile_request) {
|
|
return result.clone();
|
|
}
|
|
|
|
let path = std::env::var("CARGO_MANIFEST_DIR").unwrap() + "/../gpu-compiler/Cargo.toml";
|
|
let result = compile_request.compile(state.compile_dir.path().to_str().expect("non utf8 tempdir path"), &path).map_err(|e| {
|
|
eprintln!("compilation failed: {}", e);
|
|
StatusCode::INTERNAL_SERVER_ERROR
|
|
})?;
|
|
state.cache.write().unwrap().insert(compile_request, Ok(result.clone()));
|
|
Ok(result)
|
|
}
|