34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
"""Torch/CUDA pixel pipeline (develop, detection buffers, LUT apply). CPU parity path unchanged."""
|
|
|
|
from .device import gpu_pipeline_available, get_torch_device
|
|
from .pipeline import apply_probe_export_gpu, build_merged_lut_probe_style_gpu, color_match_one_shot_gpu, profile_gpu_pipeline
|
|
from .pipeline_full_gpu import (
|
|
apply_probe_export_full_gpu,
|
|
build_merged_lut_full_gpu,
|
|
color_match_one_shot_full_gpu,
|
|
profile_full_gpu_pipeline,
|
|
)
|
|
from .pipeline_fast import (
|
|
apply_probe_export_fast_gpu,
|
|
build_merged_lut_fast_gpu,
|
|
color_match_one_shot_fast_gpu,
|
|
profile_fast_gpu_pipeline,
|
|
)
|
|
|
|
__all__ = [
|
|
"apply_probe_export_gpu",
|
|
"apply_probe_export_full_gpu",
|
|
"apply_probe_export_fast_gpu",
|
|
"build_merged_lut_probe_style_gpu",
|
|
"build_merged_lut_full_gpu",
|
|
"build_merged_lut_fast_gpu",
|
|
"color_match_one_shot_gpu",
|
|
"color_match_one_shot_full_gpu",
|
|
"color_match_one_shot_fast_gpu",
|
|
"profile_gpu_pipeline",
|
|
"profile_full_gpu_pipeline",
|
|
"profile_fast_gpu_pipeline",
|
|
"get_torch_device",
|
|
"gpu_pipeline_available",
|
|
]
|