昇腾社区首页
中文
注册

beta)torch_npu.npu.is_jit_compile_false

接口原型

torch_npu.npu.is_jit_compile_false()

功能描述

确认算子计算是否采用的二进制,如果是二进制计算,返回True,否则返回False。

输出说明

bool型。

支持的型号

  • Atlas 训练系列产品
  • Atlas A2 训练系列产品
  • Atlas 推理系列产品

调用示例

>>>torch.npu.set_compile_mode(jit_compile=False)
>>>torch_npu.npu.is_jit_compile_false()
True