@@ -180,7 +180,7 @@ data = data.to(memory_format=torch.channels_last)
180180model = model.to("xpu")
181181data = data.to("xpu")
182182model = torch.xpu.optimize(model, dtype=torch.float32)
183- ################################## ####################
183+ #################### code changes ####################
184184
185185with torch.no_grad():
186186 model(data)
@@ -207,7 +207,7 @@ data = torch.randint(vocab_size, size=[batch_size, seq_length])
207207model = model.to("xpu")
208208data = data.to("xpu")
209209model = torch.xpu.optimize(model, dtype=torch.float32)
210- ################################## ####################
210+ #################### code changes ####################
211211
212212with torch.no_grad():
213213 model(data)
@@ -237,7 +237,7 @@ data = data.to(memory_format=torch.channels_last)
237237model = model.to("xpu")
238238data = data.to("xpu")
239239model = torch.xpu.optimize(model, dtype=torch.float32)
240- ################################## ####################
240+ #################### code changes ####################
241241
242242with torch.no_grad():
243243 d = torch.rand(1, 3, 224, 224)
@@ -271,7 +271,7 @@ data = torch.randint(vocab_size, size=[batch_size, seq_length])
271271model = model.to("xpu")
272272data = data.to("xpu")
273273model = torch.xpu.optimize(model, dtype=torch.float32)
274- ################################## ####################
274+ #################### code changes ####################
275275
276276with torch.no_grad():
277277 d = torch.randint(vocab_size, size=[batch_size, seq_length])
@@ -311,7 +311,7 @@ data = data.to(memory_format=torch.channels_last)
311311model = model.to("xpu")
312312data = data.to("xpu")
313313model = torch.xpu.optimize(model, dtype=torch.bfloat16)
314- ################################## ####################
314+ #################### code changes ####################
315315
316316with torch.no_grad():
317317 ################################# code changes ######################################
@@ -341,7 +341,7 @@ data = torch.randint(vocab_size, size=[batch_size, seq_length])
341341model = model.to("xpu")
342342data = data.to("xpu")
343343model = torch.xpu.optimize(model, dtype=torch.bfloat16)
344- ################################## ####################
344+ #################### code changes ####################
345345
346346with torch.no_grad():
347347 ################################# code changes ######################################
@@ -374,7 +374,7 @@ data = data.to(memory_format=torch.channels_last)
374374model = model.to("xpu")
375375data = data.to("xpu")
376376model = torch.xpu.optimize(model, dtype=torch.bfloat16)
377- ################################## ####################
377+ #################### code changes ####################
378378
379379with torch.no_grad():
380380 d = torch.rand(1, 3, 224, 224)
@@ -408,7 +408,7 @@ data = torch.randint(vocab_size, size=[batch_size, seq_length])
408408model = model.to("xpu")
409409data = data.to("xpu")
410410model = torch.xpu.optimize(model, dtype=torch.bfloat16)
411- ################################## ####################
411+ #################### code changes ####################
412412
413413with torch.no_grad():
414414 d = torch.randint(vocab_size, size=[batch_size, seq_length])
@@ -448,7 +448,7 @@ data = data.to(memory_format=torch.channels_last)
448448model = model.to("xpu")
449449data = data.to("xpu")
450450model = torch.xpu.optimize(model, dtype=torch.float16)
451- ################################## ####################
451+ #################### code changes ####################
452452
453453with torch.no_grad():
454454 ################################# code changes ######################################
@@ -478,7 +478,7 @@ data = torch.randint(vocab_size, size=[batch_size, seq_length])
478478model = model.to("xpu")
479479data = data.to("xpu")
480480model = torch.xpu.optimize(model, dtype=torch.float16)
481- ################################## ####################
481+ #################### code changes ####################
482482
483483with torch.no_grad():
484484 ################################# code changes ######################################
@@ -511,7 +511,7 @@ data = data.to(memory_format=torch.channels_last)
511511model = model.to("xpu")
512512data = data.to("xpu")
513513model = torch.xpu.optimize(model, dtype=torch.float16)
514- ################################## ####################
514+ #################### code changes ####################
515515
516516with torch.no_grad():
517517 d = torch.rand(1, 3, 224, 224)
@@ -545,7 +545,7 @@ data = torch.randint(vocab_size, size=[batch_size, seq_length])
545545model = model.to("xpu")
546546data = data.to("xpu")
547547model = torch.xpu.optimize(model, dtype=torch.float16)
548- ################################## ####################
548+ #################### code changes ####################
549549
550550with torch.no_grad():
551551 d = torch.randint(vocab_size, size=[batch_size, seq_length])
0 commit comments