torch_npu.npu.is_jit_compile_false()
确认算子计算是否采用的二进制,如果是二进制计算,返回True,否则返回False。
bool型。
>>>torch.npu.set_compile_mode(jit_compile=False) >>>torch_npu.npu.is_jit_compile_false() True