ITADN

JIT-compilation fails

#516Openjulcst 创建于 2025-09-06
J
julcstcommented
Thank you for this great library, I recently I tried out the new JIT-fusion mode however it fails with this message: ``` tiny-cuda-nn warning: duplicate named arg Failed to JIT-compile `training_step_network_with_input_encoding`. Disabling JIT. ``` My input size is 14, my output size is 3, my batch size 256 * 64 * 8 and my configuration: ```json { "encoding": { "otype": "Composite", "nested": [ { "n_dims_to_encode": 3, "otype": "Grid", "type": "Hash", "n_levels": 16, "n_features_per_level": 2, "log2_hashmap_size": 19, "base_resolution": 16, "per_level_scale": 2.0, "interpolation": "Linear" }, { "n_dims_to_encode": 5, "otype": "OneBlob", "n_bins": 4 }, { "otype": "Identity" } ] }, "network": { "otype": "FullyFusedMLP", "activation": "ReLU", "output_activation": "None", "n_neurons": 64, "n_hidden_layers": 2 }, "loss": { "otype": "RelativeL2Luminance" }, "optimizer": { "otype": "EMA", "decay": 0.95, "nested": { "otype": "Adam", "learning_rate": 1e-2, "beta1": 0.9, "beta2": 0.99, "epsilon": 1e-15, "l2_reg": 1e-6, "absolute_decay": 0, "relative_decay": 0, "adabound": false } } } ``` Does someone know what could be going wrong here? I run my tests on C++20 using GCC14, CudaToolkit 12.8.93, Fedora Linux and an RTX 3060 Ti. Any help appreciated
1 条评论