diff options
Diffstat (limited to 'src/gpu/effects/shaders.cc')
| -rw-r--r-- | src/gpu/effects/shaders.cc | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/src/gpu/effects/shaders.cc b/src/gpu/effects/shaders.cc index 625c5b6..6559bf5 100644 --- a/src/gpu/effects/shaders.cc +++ b/src/gpu/effects/shaders.cc @@ -51,6 +51,13 @@ void InitShaderComposer() { register_if_exists("ray_box", AssetId::ASSET_SHADER_RAY_BOX); register_if_exists("ray_triangle", AssetId::ASSET_SHADER_RAY_TRIANGLE); + + register_if_exists("cnn_activation", AssetId::ASSET_SHADER_CNN_ACTIVATION); + register_if_exists("cnn_conv3x3", AssetId::ASSET_SHADER_CNN_CONV3X3); + register_if_exists("cnn_conv5x5", AssetId::ASSET_SHADER_CNN_CONV5X5); + register_if_exists("cnn_conv7x7", AssetId::ASSET_SHADER_CNN_CONV7X7); + register_if_exists("cnn_weights_generated", + AssetId::ASSET_SHADER_CNN_WEIGHTS); } // Helper to get asset string or empty string @@ -99,6 +106,10 @@ const char* chroma_aberration_shader_wgsl = SafeGetAsset(AssetId::ASSET_SHADER_CHROMA_ABERRATION); +const char* cnn_layer_shader_wgsl = + + SafeGetAsset(AssetId::ASSET_SHADER_CNN_LAYER); + const char* gen_noise_compute_wgsl = SafeGetAsset(AssetId::ASSET_SHADER_COMPUTE_GEN_NOISE); |