1
1
Intelยฎ Extension for PyTorch* ๋ฐฑ์๋
2
2
=====================================
3
3
4
- **์ ์ **: `Hamid Shojanazeri <https://github.com/jingxu10>__ `
5
- **๋ฒ์ญ: **: `๊น์ฌํ <https://github.com/jh941213>__ `
4
+ **์ ์ **: `Hamid Shojanazeri <https://github.com/jingxu10 >`_
5
+ **๋ฒ์ญ: **: `๊น์ฌํ <https://github.com/jh941213 >`_
6
6
7
7
- `torch.compile ` ๊ณผ ๋ ์ ์๋ํ๋๋ก, Intelยฎ Extension for PyTorch๋ ``ipex `` ๋ผ๋ ๋ฐฑ์๋๋ฅผ ๊ตฌํํ์ต๋๋ค.
8
8
- ์ด ๋ฐฑ์๋๋ Intel ํ๋ซํผ์์ ํ๋์จ์ด ์์ ์ฌ์ฉ ํจ์จ์ฑ์ ๊ฐ์ ํ์ฌ ์ฑ๋ฅ์ ํฅ์์ํค๋ ๊ฒ์ ๋ชฉํ๋ก ํฉ๋๋ค.
@@ -47,7 +47,6 @@ FP32 ํ์ต
47
47
48
48
# ################### ์ฝ๋ ๋ณ๊ฒฝ ๋ถ๋ถ ####################
49
49
import intel_extension_for_pytorch as ipex
50
-
51
50
# ์ ํ์ ์ผ๋ก ๋ค์ API๋ฅผ ํธ์ถํ์ฌ, ํ๋ก ํธ์๋ ์ต์ ํ๋ฅผ ์ ์ฉํฉ๋๋ค.
52
51
model, optimizer = ipex.optimize(model, optimizer = optimizer)
53
52
@@ -99,7 +98,6 @@ BF16 ํ์ต
99
98
100
99
# ################### ์ฝ๋ ๋ณ๊ฒฝ ๋ถ๋ถ ####################
101
100
import intel_extension_for_pytorch as ipex
102
-
103
101
# ์ ํ์ ์ผ๋ก ๋ค์ API๋ฅผ ํธ์ถํ์ฌ, ํ๋ก ํธ์๋ ์ต์ ํ๋ฅผ ์ ์ฉํฉ๋๋ค.
104
102
model, optimizer = ipex.optimize(model, dtype = torch.bfloat16, optimizer = optimizer)
105
103
@@ -131,7 +129,6 @@ FP32 ์ถ๋ก
131
129
132
130
# ################### ์ฝ๋ ๋ณ๊ฒฝ ๋ถ๋ถ ####################
133
131
import intel_extension_for_pytorch as ipex
134
-
135
132
# ์ ํ์ ์ผ๋ก ๋ค์ API๋ฅผ ํธ์ถํ์ฌ, ํ๋ก ํธ์๋ ์ต์ ํ๋ฅผ ์ ์ฉํฉ๋๋ค.
136
133
model = ipex.optimize(model, weights_prepack = False )
137
134
@@ -158,7 +155,6 @@ BF16 ์ถ๋ก
158
155
159
156
# ################### ์ฝ๋ ๋ณ๊ฒฝ ๋ถ๋ถ ####################
160
157
import intel_extension_for_pytorch as ipex
161
-
162
158
# ์ ํ์ ์ผ๋ก ๋ค์ API๋ฅผ ํธ์ถํ์ฌ, ํ๋ก ํธ์๋ ์ต์ ํ๋ฅผ ์ ์ฉํฉ๋๋ค.
163
159
model = ipex.optimize(model, dtype = torch.bfloat16, weights_prepack = False )
164
160
0 commit comments