Add compile backend arg for test_set_compiler_fn

Add compile backend arg for test_set_compiler_fn #6451