code: def triton_causal_fw_bw(x):
      y = torch.norm(triton_softmax(x, causal=True))
      y.backward()
  
file_path: xformers/xformers/benchmarks/benchmark_triton_softmax.py
imports: import torch
  from xformers.benchmarks.utils import TestCase, bench_functions
  from xformers.triton.softmax import log_softmax
  from xformers.triton.softmax import softmax
label_desc: Use image node_id 4 for calling a global function with example usage: triton_causal_fw_bw(x) without return types