return Trueifcan_fuse_into_triton_kernel_wrapper_functional(a,b):returnTrue
+ if(
+ a.targetisoperator.getitem
+ anda.args[0].target
+ istorch.ops.higher_order.triton_kernel_wrapper_functional
+ ):
+ # if a is the output of a user triton kernel,
+ # then (by default) we will not be able to fuse b into it
+ returnFalsereturnop_types.is_fusible(a)andop_types.is_fusible(b)try:
diff --git a/nightly/aot_autograd.html b/nightly/aot_autograd.html
index 99383932a..c89a6f82a 100644
--- a/nightly/aot_autograd.html
+++ b/nightly/aot_autograd.html
@@ -267,7 +267,7 @@