[functorch] add conditional functionalization (pytorch/functorch#235)
* add conditional functionalization
* fix an RAII bug </3
* fix an RAII bug </3
* remove swap files
* also functionalize block_diag
* add a vmap(vmap(diag_embed)) test, update usage of sync() in the conditional fallback logic