Commits


Vincent Wang authored and GitHub committed 1bfc26685b5
ATen Op Supports Int Return Type and CPU Tensor Arguments (#19773) This PR: - add support for int as return type, will create a CPU scalar tensor for it. - add attributes to specify which arguments or returns are CPU tensors. - adjust ATen efficient attn to match latest PyTorch native function. - a Triton codegen bugfix by the way.