Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

tweak on metal ci #414

Merged
merged 11 commits into from
Jan 12, 2025
Merged

tweak on metal ci #414

Changes from 1 commit
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
86baa83
runtime workloads
hikettei Jan 11, 2025
d0786e3
restrict range
hikettei Jan 11, 2025
eb5b1c3
with-torch uses elements
hikettei Jan 11, 2025
f1a26ba
MOD FLOPs
hikettei Jan 11, 2025
69b3ef9
Fix activation tests
hikettei Jan 11, 2025
3b5dfb7
Disable auto parallelize for now
hikettei Jan 11, 2025
0594747
cast
hikettei Jan 11, 2025
6ff1421
Tweak FLT_MIN
hikettei Jan 11, 2025
d2f25f2
smolify the range
hikettei Jan 12, 2025
9aafef0
smol sin test
hikettei Jan 12, 2025
45db97d
Fix: Softmax tests for mtl
hikettei Jan 12, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
cast
  • Loading branch information
@hikettei
hikettei committed Jan 11, 2025
commit 0594747fb2571f1701aa7c277a46ade37cb6983b
9 changes: 6 additions & 3 deletions source/test-suite/test-apis.lisp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -480,7 +480,10 @@
(ok (every (equal-to 4) (elements a))))))

(defclass TestIndexComponents (Func) nil)
(defmethod forward ((op TestIndexComponents) &rest inputs) (st "A[~] -> A[~]" ((car inputs))))
(defmethod forward ((op TestIndexComponents) &rest inputs)
(let ((out (st "A[~] -> A[~]" ((car inputs)))))
(setf (tensor-dtype out) :int64)
out))
(defmethod backward ((op TestIndexComponents) &optional dout) dout)
(defmethod lower ((op TestIndexComponents) &rest inputs)
(with-context (_ (%index-components (car inputs) (cdr inputs)))))
Expand All @@ -495,8 +498,8 @@
(elements (proceed (test-ic (make-tensor `(4 5))))))
"Does symbolic index-component work?")
(ok (every #'=
(elements (pproceed `((a . 4) (b . 5)) (!sin (test-ic (make-tensor `(a b))))))
(elements (proceed (!sin (test-ic (make-tensor `(4 5)))))))
(elements (pproceed `((a . 4) (b . 5)) (!sin (!cast (test-ic (make-tensor `(a b))) :float32))))
(elements (proceed (!sin (!cast (test-ic (make-tensor `(4 5))) :float32)))))
"Fused with Unary")))

(deftest threefry2x32
Expand Down
Loading