pytorch

fd553c46 - nvprim op support runtime checks on dtype compatibility on prims.convert_element_type (#85566)

nvprim op support runtime checks on dtype compatibility on prims.convert_element_type (#85566) I'm seeing issue that we lower `_to_copy` into `nvprims.convert_element_type`. In cases where we are casting to a dtype that's not supported by nvfuser, this raise runtime error. I added a quick check in the lowering part where each op can peek at fx.node and make a runtime decision on whether the given op should be lowered to nvprim. Pull Request resolved: https://github.com/pytorch/pytorch/pull/85566 Approved by: https://github.com/IvanYashchuk, https://github.com/ngimel