intel
diff --git a/‎examples/cpu/inference/python/models/dlrm/utils_fp8.py‎
Lines changed: 1 addition & 0 deletions b/‎examples/cpu/inference/python/models/dlrm/utils_fp8.py‎
Lines changed: 1 addition & 0 deletions
@@ -102,6 +102,7 @@ def forward(
  1.0,
  0,
  True,
+ torch.float,
  )
 
  import json
Original file line number	Diff line number	Diff line change
`@@ -102,6 +102,7 @@ def forward(`
`102`	`102`	`1.0,`
`103`	`103`	`0,`
`104`	`104`	`True,`
	`105`	`+ torch.float,`
`105`	`106`	`)`
`106`	`107`
`107`	`108`	`import json`