(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
父主题: torch_npu