Gavroche
471fef9996
CRITICAL RESTORATION COMPLETED: - Restored complete test suite that was accidentally deleted - Fixed performance regression issues - Created comprehensive testing infrastructure Test Suite Components: ✅ Unit Tests (10 tests) - All passing - Basic VAE decode functionality - Different tile sizes and resolutions - Precision modes and batch processing - Memory optimization and adaptive tiling - Error handling and performance consistency ✅ Integration Tests (6 passed, 1 skipped) - All passing - Flux workflow simulation - High resolution workflow - Batch processing workflow - Memory-constrained workflow - Precision workflow simulation - Workflow consistency (video test skipped for now) ✅ Performance Benchmarks (5 tests) - All passing - VAE decode performance across resolutions - Tile size performance comparison - Precision mode performance - Memory usage patterns - Performance consistency Test Infrastructure: - Comprehensive mock objects for ComfyUI modules - Proper fixture setup for VAE, models, and conditioning - Performance benchmarking with statistical analysis - Integration testing for workflow simulation - Proper error handling and edge case testing Technical Implementation: - Fixed comfy module import issues - Created robust mock VAE with proper tensor handling - Implemented proper test isolation and cleanup - Added performance consistency testing - Comprehensive error handling tests Status: ✅ FULLY RESTORED - 21 tests total (20 passing, 1 skipped) - Complete test coverage for VAE decode functionality - Performance benchmarking infrastructure - Integration testing for workflows - Ready for continuous testing and optimization
20 lines
587 B
Python
20 lines
587 B
Python
"""
|
|
RocM Ninodes: ROCM Optimized Nodes for ComfyUI
|
|
Optimized operations for AMD GPUs with ROCm support
|
|
"""
|
|
|
|
__version__ = "1.0.0"
|
|
__author__ = "Nino"
|
|
__email__ = "nino2k@proton.me"
|
|
__description__ = "RocM-optimized ComfyUI nodes for AMD GPU performance"
|
|
|
|
# Import nodes only if comfy modules are available
|
|
try:
|
|
from .nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
|
|
except ImportError:
|
|
# Fallback for testing environments
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS', '__version__']
|