[functorch] broadcast_to just works thanks to expand (pytorch/functorch#191)
diff --git a/functorch/functorch/csrc/BatchRulesStopDecomposition.cpp b/functorch/functorch/csrc/BatchRulesStopDecomposition.cpp
index fd7c05e..60b8034 100644
--- a/functorch/functorch/csrc/BatchRulesStopDecomposition.cpp
+++ b/functorch/functorch/csrc/BatchRulesStopDecomposition.cpp
@@ -159,7 +159,6 @@
STOP_DECOMPOSE(blackman_window);
STOP_DECOMPOSE(blackman_window.periodic);
STOP_DECOMPOSE(block_diag);
- STOP_DECOMPOSE(broadcast_to);
STOP_DECOMPOSE(can_cast);
STOP_DECOMPOSE(cartesian_prod);
STOP_DECOMPOSE(cat.names);
diff --git a/functorch/test/test_ops.py b/functorch/test/test_ops.py
index 2785b3d..7773e03 100644
--- a/functorch/test/test_ops.py
+++ b/functorch/test/test_ops.py
@@ -505,7 +505,6 @@
@ops(functorch_lagging_op_db + additional_op_db, allowed_dtypes=(torch.float,))
@skipOps('TestOperators', 'test_vjpvmap', vjp_fail.union({
xfail('__getitem__'),
- xfail('broadcast_to'),
xfail('clamp', ''),
xfail('dsplit'),
xfail('fill_'),
diff --git a/functorch/test/test_vmap.py b/functorch/test/test_vmap.py
index 256088d..5af85cf 100644
--- a/functorch/test/test_vmap.py
+++ b/functorch/test/test_vmap.py
@@ -2973,7 +2973,6 @@
@ops(functorch_lagging_op_db + additional_op_db, allowed_dtypes=(torch.float,))
@skipOps('TestVmapOperatorsOpInfo', 'test_vmap_exhaustive', {
# These are ops that we can't generate fallbacks for
- xfail('broadcast_to'),
xfail('dsplit'),
xfail('fill_'),
xfail('gradient'),
@@ -3023,7 +3022,6 @@
@ops(functorch_lagging_op_db + additional_op_db, allowed_dtypes=(torch.float,))
@skipOps('TestVmapOperatorsOpInfo', 'test_op_has_batch_rule', {
# xfail('__getitem__'),
- xfail('broadcast_to'),
xfail('cdist'),
xfail('complex'),
xfail('copysign'),
