Thanks for sending an issue! Please fill in the following template to help quickly solve your problem.
mask/x_clamped_loss dtype 校验与 dtype 组合联动拦截不到位,需要新增
Ascend950
TilingFunc GetShapeAndDtype 原仅校验 y_grad dtype,mask/x_clamped_loss dtype 及 dtype 组合合法性未校验,非法 dtype(如 mask=int32)和非法组合(如 fp16/fp32/fp16)经 PrecisionReduceFlag Cast 后可绕过约束进入 kernel。
增补:
TTK kernel 模式 Ascend950 实跑验证:3 个应拦截未拦截 probe 转为 INVALID_TILING 有效拦截,2 个合法 control 仍 PASS。
无
Thanks for sending an issue! Please fill in the following template to help quickly solve your problem.
Describe the current behavior / 问题描述 (Mandatory / 必填)
mask/x_clamped_loss dtype 校验与 dtype 组合联动拦截不到位,需要新增
Environment / 环境信息 (Mandatory / 必填)
Ascend950
Steps to reproduce the issue / 重现步骤 (Mandatory / 必填)
TilingFunc GetShapeAndDtype 原仅校验 y_grad dtype,mask/x_clamped_loss dtype
及 dtype 组合合法性未校验,非法 dtype(如 mask=int32)和非法组合(如
fp16/fp32/fp16)经 PrecisionReduceFlag Cast 后可绕过约束进入 kernel。
增补:
(对应 def.cpp 4 列合法 profile)
Describe the expected behavior / 预期结果 (Mandatory / 必填)
TTK kernel 模式 Ascend950 实跑验证:3 个应拦截未拦截 probe 转为 INVALID_TILING
有效拦截,2 个合法 control 仍 PASS。
Related log / screenshot / 日志 / 截图 (Mandatory / 必填)
无
Special notes for this issue/备注 (Optional / 选填)