fork-3007-hit2sjtu/dev - Branch - CodSpeedHQ/kornia

Fix type mis-match between out and patches during mixed-precision training(#86)

Comparing

fork-3007-hit2sjtu/dev

(

2737749

) with

main

(

d645a84

)

test_aug_2d_thin_plate_spline[1-1-128-128-cpu-float32-inductor]

benchmarks/augmentation/2d_geometric_test.py::test_aug_2d_thin_plate_spline[1-1-128-128-cpu-float32-inductor]

7.8 s

test_aug_2d_elastic_transform[1-1-128-128-cpu-float32-inductor]

benchmarks/augmentation/2d_geometric_test.py::test_aug_2d_elastic_transform[1-1-128-128-cpu-float32-inductor]

8.5 s

test_aug_2d_solarize[1-1-128-128-cpu-float32-inductor]

benchmarks/augmentation/2d_intensity_test.py::test_aug_2d_solarize[1-1-128-128-cpu-float32-inductor]

3 s

test_aug_2d_linear_corner_illumination[1-1-128-128-cpu-float32-inductor]

benchmarks/augmentation/2d_intensity_test.py::test_aug_2d_linear_corner_illumination[1-1-128-128-cpu-float32-inductor]

-1%

4.6 s

Click on a commit to change the comparison range

base

main

d645a84

Fix type mis-match during mixed-precision training. Out and patches can be (Float Half), thus cause issue. Signed-off-by: Qiang Wang <wq@sightp.com>

2737749

2 months ago

by hit2sjtu