15427c96c0
- Change file mode from 644 to 755 for all project files - Add chain_id parameter to get_balance RPC endpoint with default "ait-devnet" - Rename Miner.extra_meta_data to extra_metadata for consistency
59 lines
1.6 KiB
Python
Executable File
59 lines
1.6 KiB
Python
Executable File
"""
|
|
Refactored FastAPI GPU Acceleration Service
|
|
|
|
Uses the new abstraction layer for backend-agnostic GPU acceleration.
|
|
"""
|
|
|
|
from fastapi import FastAPI, HTTPException
|
|
from pydantic import BaseModel
|
|
from typing import Dict, List, Optional
|
|
import logging
|
|
|
|
from .gpu_manager import GPUAccelerationManager, create_gpu_manager
|
|
|
|
app = FastAPI(title="AITBC GPU Acceleration API")
|
|
logger = logging.getLogger(__name__)
|
|
|
|
# Initialize GPU manager
|
|
gpu_manager = create_gpu_manager()
|
|
|
|
class FieldOperation(BaseModel):
|
|
a: List[int]
|
|
b: List[int]
|
|
|
|
class MultiScalarOperation(BaseModel):
|
|
scalars: List[List[int]]
|
|
points: List[List[int]]
|
|
|
|
@app.post("/field/add")
|
|
async def field_add(op: FieldOperation):
|
|
"""Perform field addition."""
|
|
try:
|
|
a = np.array(op.a, dtype=np.uint64)
|
|
b = np.array(op.b, dtype=np.uint64)
|
|
result = gpu_manager.field_add(a, b)
|
|
return {"result": result.tolist()}
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=str(e))
|
|
|
|
@app.post("/field/mul")
|
|
async def field_mul(op: FieldOperation):
|
|
"""Perform field multiplication."""
|
|
try:
|
|
a = np.array(op.a, dtype=np.uint64)
|
|
b = np.array(op.b, dtype=np.uint64)
|
|
result = gpu_manager.field_mul(a, b)
|
|
return {"result": result.tolist()}
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=str(e))
|
|
|
|
@app.get("/backend/info")
|
|
async def backend_info():
|
|
"""Get backend information."""
|
|
return gpu_manager.get_backend_info()
|
|
|
|
@app.get("/performance/metrics")
|
|
async def performance_metrics():
|
|
"""Get performance metrics."""
|
|
return gpu_manager.get_performance_metrics()
|